Ayush0804 commited on
Commit
781befc
Β·
verified Β·
1 Parent(s): 15f516d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -12,8 +12,7 @@ model_id="microsoft/Phi-3.5-vision-instruct"
12
  model=AutoModelForCausalLM.from_pretrained(
13
  model_id,
14
  trust_remote_code=True,
15
- torch_dtype=torch.float16,
16
- use_flash_attention_2=False)
17
 
18
  processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True, num_crops=16)
19
  math_messages=[]
 
12
  model=AutoModelForCausalLM.from_pretrained(
13
  model_id,
14
  trust_remote_code=True,
15
+ torch_dtype=torch.float16,)
 
16
 
17
  processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True, num_crops=16)
18
  math_messages=[]