Skip to content

Commit 931b465

Browse files
committed
fix phi3-vision bug
1 parent 07de89f commit 931b465

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

swift/llm/utils/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1160,7 +1160,7 @@ def get_model_tokenizer_paligemma_vision(model_dir: str,
11601160
return model, tokenizer
11611161

11621162

1163-
def _clone_grads(module, input, output):
1163+
def _clone_hook(module, input, output):
11641164
return output.requires_grad_(True).clone()
11651165

11661166

@@ -1184,7 +1184,7 @@ def get_model_tokenizer_phi3_vision(model_dir: str,
11841184
tokenizer.processor = processor
11851185

11861186
if load_model:
1187-
model.model.vision_embed_tokens.wte.register_forward_hook(_clone_grads)
1187+
model.model.vision_embed_tokens.wte.register_forward_hook(_clone_hook)
11881188

11891189
return model, tokenizer
11901190

0 commit comments

Comments
 (0)