mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-06 00:57:33 +08:00
fix (#3640)
This commit is contained in:
@@ -257,7 +257,7 @@ class Qwen3ForCausalLM(ModelForCasualLM):
|
|||||||
"""
|
"""
|
||||||
self.model.load_state_dict(state_dict)
|
self.model.load_state_dict(state_dict)
|
||||||
if self.tie_word_embeddings:
|
if self.tie_word_embeddings:
|
||||||
self.lm_head.load_state_dict({self.lm_head.weight_key: self.ernie.embed_tokens.embeddings.weight})
|
self.lm_head.load_state_dict({self.lm_head.weight_key: self.model.embed_tokens.embeddings.weight})
|
||||||
else:
|
else:
|
||||||
self.lm_head.load_state_dict(state_dict)
|
self.lm_head.load_state_dict(state_dict)
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user