andito HF staff commited on
Commit
0a1a1a0
1 Parent(s): 0a651e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,9 +9,9 @@ import subprocess
9
  #subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
10
 
11
 
12
- processor = AutoProcessor.from_pretrained("HuggingFaceTB/SmolVLM_converted_4")
13
 
14
- model = Idefics3ForConditionalGeneration.from_pretrained("HuggingFaceTB/SmolVLM_converted_4",
15
  torch_dtype=torch.bfloat16,
16
  #_attn_implementation="flash_attention_2"
17
  ).to("cuda")
 
9
  #subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
10
 
11
 
12
+ processor = AutoProcessor.from_pretrained("HuggingFaceTB/SmolVLM-Instruct")
13
 
14
+ model = Idefics3ForConditionalGeneration.from_pretrained("HuggingFaceTB/SmolVLM-Instruct",
15
  torch_dtype=torch.bfloat16,
16
  #_attn_implementation="flash_attention_2"
17
  ).to("cuda")