arcma commited on
Commit
2146041
·
1 Parent(s): a1b736d

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -18,7 +18,7 @@ def check(x):
18
  return False
19
  return True
20
 
21
- @torch.jit.script
22
  def process_image(pixel_values):
23
  with torch.no_grad():
24
  generated_ids = model.generate(pixel_values, num_beams=1, num_return_sequences=1)
 
18
  return False
19
  return True
20
 
21
+ #@torch.jit.script
22
  def process_image(pixel_values):
23
  with torch.no_grad():
24
  generated_ids = model.generate(pixel_values, num_beams=1, num_return_sequences=1)