SmolVLM-256M-Demo

Running on Zero

andito HF staff commited on 7 days ago

Commit

0b32b82

verified ·

1 Parent(s): 6d64276

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,8 +10,8 @@ import spaces
 #subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
-processor = AutoProcessor.from_pretrained("HuggingFaceTB/SmolVLM-Instruct")
-model = AutoModelForVision2Seq.from_pretrained("HuggingFaceTB/SmolVLM-Instruct",
         torch_dtype=torch.bfloat16,
         #_attn_implementation="flash_attention_2"
         ).to("cuda")

 #subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+processor = AutoProcessor.from_pretrained("HuggingFaceTB/SmolVLM-Instruct-250M")
+model = AutoModelForVision2Seq.from_pretrained("HuggingFaceTB/SmolVLM-Instruct-250M",
         torch_dtype=torch.bfloat16,
         #_attn_implementation="flash_attention_2"
         ).to("cuda")