We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f52e327 commit b37c603Copy full SHA for b37c603
swift/llm/model/patcher.py
@@ -74,7 +74,7 @@ def lm_head_forward(self, hidden_states):
74
return hidden_states
75
76
lm_heads = ['lm_head', 'output', 'embed_out', 'output_layer']
77
- llm_model = get_llm_model(module)
+ llm_model = get_llm_model(module, model_meta=model_meta)
78
79
found = False
80
for lm_head in lm_heads:
0 commit comments