Tiny refactor DefaultModelLoader.Source (#5482)

2025-04-28 15:35:51 +08:00
parent 3029889cb4
commit 644ed409d1
2 changed files with 11 additions and 13 deletions
--- a/python/sglang/srt/model_executor/model_runner.py
+++ b/python/sglang/srt/model_executor/model_runner.py
@@ -515,13 +515,7 @@ class ModelRunner:

        def get_weight_iter(config):
            iter = loader._get_weights_iterator(
-                DefaultModelLoader.Source(
-                    config.model_path,
-                    revision=config.revision,
-                    fall_back_to_pt=getattr(
-                        self.model, "fall_back_to_pt_during_load", True
-                    ),
-                )
+                DefaultModelLoader.Source.init_new(config, model)
            )
            return iter