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 8a49299 commit 7b57fb2Copy full SHA for 7b57fb2
swift/llm/utils/model.py
@@ -2628,6 +2628,7 @@ def get_model_tokenizer_florence(model_dir: str,
2628
else:
2629
model_config.text_config._flash_attn_2_enabled = use_flash_attn
2630
model_config._flash_attn_2_enabled = use_flash_attn
2631
+ model_config.vision_config.model_type = 'davit'
2632
model, tokenizer = get_model_tokenizer_from_repo(
2633
model_dir,
2634
torch_dtype,
0 commit comments