sagar007 commited on
Commit
5b2b4ac
1 Parent(s): 32f55fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -7,16 +7,14 @@ import torch
7
 
8
  # Load the model and processor
9
  model_id = "microsoft/Phi-3.5-vision-instruct"
10
- model_revision = "f2a2b357af3e062d60ca6e73a13f9f97a7fd3524" # Pin to a specific revision
11
 
12
  model = AutoModelForCausalLM.from_pretrained(
13
  model_id,
14
- revision=model_revision,
15
  trust_remote_code=True,
16
  torch_dtype=torch.float16,
17
  use_flash_attention_2=False, # Explicitly disable Flash Attention 2
18
  )
19
- processor = AutoProcessor.from_pretrained(model_id, revision=model_revision, trust_remote_code=True, num_crops=16)
20
 
21
  @spaces.GPU(duration=120) # Adjust the duration as needed
22
  def solve_math_problem(image):
 
7
 
8
  # Load the model and processor
9
  model_id = "microsoft/Phi-3.5-vision-instruct"
 
10
 
11
  model = AutoModelForCausalLM.from_pretrained(
12
  model_id,
 
13
  trust_remote_code=True,
14
  torch_dtype=torch.float16,
15
  use_flash_attention_2=False, # Explicitly disable Flash Attention 2
16
  )
17
+ processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True, num_crops=16)
18
 
19
  @spaces.GPU(duration=120) # Adjust the duration as needed
20
  def solve_math_problem(image):