merve HF staff commited on
Commit
6bc5bc2
1 Parent(s): 9ad1205

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-new")
13
 
14
- model = Idefics3ForConditionalGeneration.from_pretrained("HuggingFaceM4/idefics3-8b-new",
15
  torch_dtype=torch.bfloat16,
16
  #_attn_implementation="flash_attention_2",
17
  trust_remote_code=True).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("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
  trust_remote_code=True).to("cuda")