andito HF staff commited on
Commit
d24c16c
1 Parent(s): 5d9cb65

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("HuggingFaceM4/Idefics3-8B-Llama3")
13
 
14
- model = Idefics3ForConditionalGeneration.from_pretrained("HuggingFaceM4/Idefics3-8B-Llama3",
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_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")