Skip to content

Commit b37c603

Browse files
tastelikefeettastelikefeet
and
tastelikefeet
authored
fix model_meta (#4431)
Co-authored-by: tastelikefeet <tastetastelikefeetlikefeet@B-Q60PQ05P-0329.local>
1 parent f52e327 commit b37c603

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

swift/llm/model/patcher.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def lm_head_forward(self, hidden_states):
7474
return hidden_states
7575

7676
lm_heads = ['lm_head', 'output', 'embed_out', 'output_layer']
77-
llm_model = get_llm_model(module)
77+
llm_model = get_llm_model(module, model_meta=model_meta)
7878

7979
found = False
8080
for lm_head in lm_heads:

0 commit comments

Comments
 (0)