Skip to content

Commit 7b57fb2

Browse files
authored
fix florence models(#1301)
1 parent 8a49299 commit 7b57fb2

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

swift/llm/utils/model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2628,6 +2628,7 @@ def get_model_tokenizer_florence(model_dir: str,
26282628
else:
26292629
model_config.text_config._flash_attn_2_enabled = use_flash_attn
26302630
model_config._flash_attn_2_enabled = use_flash_attn
2631+
model_config.vision_config.model_type = 'davit'
26312632
model, tokenizer = get_model_tokenizer_from_repo(
26322633
model_dir,
26332634
torch_dtype,

0 commit comments

Comments
 (0)