Hugging Face's logo

Spaces:
prithivMLmods
/
Tiny-VLMs-Lab
Running on Zero

App Files Community
12
prithivMLmods commited on
Commit
289e995
·
verified ·
1 Parent(s): 51443ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -78,7 +78,7 @@ model_g = Qwen2_5_VLForConditionalGeneration.from_pretrained(
78
 
79
  MODEL_ID_I = "HuggingFaceTB/SmolVLM2-2.2B-Instruct"
80
  processor_i = AutoProcessor.from_pretrained(MODEL_ID_I, trust_remote_code=True)
81
- model_i = AutoModelForImageTextToText.from_pretrained(
82
  MODEL_ID_I, trust_remote_code=True, torch_dtype=torch.bfloat16, _attn_implementation="flash_attention_2"
83
  ).to(device).eval()
84
 
 
78
 
79
  MODEL_ID_I = "HuggingFaceTB/SmolVLM2-2.2B-Instruct"
80
  processor_i = AutoProcessor.from_pretrained(MODEL_ID_I, trust_remote_code=True)
81
+ model_i = AutoModelForVision2Seq.from_pretrained(
82
  MODEL_ID_I, trust_remote_code=True, torch_dtype=torch.bfloat16, _attn_implementation="flash_attention_2"
83
  ).to(device).eval()
84