MaziyarPanahi commited on
Commit
bd7ed6a
·
verified ·
1 Parent(s): 482d1c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import torch
9
  from PIL import Image
10
 
11
  models = {
12
- "microsoft/Phi-3.5-vision-instruct": AutoModelForCausalLM.from_pretrained("microsoft/Phi-3.5-vision-instruct", trust_remote_code=True, torch_dtype="auto", _attn_implementation="flash_attention_2").cuda().eval()
13
  }
14
 
15
  processors = {
 
9
  from PIL import Image
10
 
11
  models = {
12
+ "microsoft/Phi-3.5-vision-instruct": AutoModelForCausalLM.from_pretrained("microsoft/Phi-3.5-vision-instruct", trust_remote_code=True, torch_dtype="auto").cuda().eval()
13
  }
14
 
15
  processors = {