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)
|