Update run.py
Browse files
run.py
CHANGED
|
@@ -18,7 +18,7 @@ def check(x):
|
|
| 18 |
return False
|
| 19 |
return True
|
| 20 |
|
| 21 |
-
|
| 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)
|