Update README.md

README.md

@@ -143,8 +143,10 @@ model_name = "deepseek-ai/DeepSeek-V3.1-Terminus"
 model, tokenizer,_=llm_load_model(model_name,trust_remote_code=False,device="cpu")
 layer_config = {}
 for n, m in model.named_modules():
+    if isinstance(m,torch.nn.Embedding)：
+        layer_config[n] = {"bits": 8}
     if isinstance(m, torch.nn.Linear):
-        if isinstance(m,torch.nn.Embedding) or n=="lm_head":
+        if n=="lm_head":
             layer_config[n] = {"bits": 8}
             continue
         if "expert" in n and "shared_experts" not in n: