load draft model fix (#7506)

8aa5ae6b · yilian49 · GitHub · 8a323557 · 8aa5ae6b
Unverified Commit 8aa5ae6b authored Jul 18, 2025 by yilian49 Committed by GitHub Jul 17, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 1 deletion

python/sglang/srt/model_loader/loader.py python/sglang/srt/model_loader/loader.py +7 -1

No files found.
--- a/python/sglang/srt/model_loader/loader.py
+++ b/python/sglang/srt/model_loader/loader.py
@@ -575,7 +575,13 @@ class DummyModelLoader(BaseModelLoader):
            # 2. Post-processing of weights, including assigning specific member variables.
            # For `dummy_init`, only the second stage is required.
            if hasattr(model, "post_load_weights"):
-                model.post_load_weights()
+                if (
+                    model_config.hf_config.architectures[0]
+                    == "DeepseekV3ForCausalLMNextN"
+                ):
+                    model.post_load_weights(is_nextn=True)
+                else:
+                    model.post_load_weights()

        return model.eval()