misc: add pre-commit config (#637)

2024-07-18 04:55:39 +10:00
parent a8552cb18b
commit 2e341cd493
43 changed files with 481 additions and 299 deletions
--- a/python/sglang/srt/models/qwen2.py
+++ b/python/sglang/srt/models/qwen2.py
@@ -313,7 +313,10 @@ class Qwen2ForCausalLM(nn.Module):
                param = params_dict[name]
                weight_loader = getattr(param, "weight_loader", default_weight_loader)
                weight_loader(param, loaded_weight)
-                if self.config.tie_word_embeddings and name=="model.embed_tokens.weight":
+                if (
+                    self.config.tie_word_embeddings
+                    and name == "model.embed_tokens.weight"
+                ):
                    weight_loader(params_dict["lm_head.weight"], loaded_weight)