Fix qwen accuracy issue (#2945)

This commit is contained in:
Ke Bao
2025-01-17 22:35:26 +08:00
committed by GitHub
parent 5dc54f1a62
commit 53e6552fed

View File

@@ -356,8 +356,6 @@ class Qwen2ForCausalLM(nn.Module):
break
else:
# Skip loading extra bias for GPTQ models.
if "lm_head.weight" in name:
continue
if name.endswith(".bias") and name not in params_dict:
continue
param = params_dict[name]