Nymbo commited on
Commit
1fda4ef
·
verified ·
1 Parent(s): 467c815

Update app.py

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