Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def perform_ocr(image):
|
|
47 |
return_tensors="pt",
|
48 |
)
|
49 |
inputs = inputs.to("cuda")
|
50 |
-
generated_ids = model.generate(**inputs, max_new_tokens=max_tokens)
|
51 |
generated_ids_trimmed = [
|
52 |
out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
|
53 |
]
|
|
|
47 |
return_tensors="pt",
|
48 |
)
|
49 |
inputs = inputs.to("cuda")
|
50 |
+
generated_ids = model.generate(**inputs, max_new_tokens=max_tokens, use_cache = True)
|
51 |
generated_ids_trimmed = [
|
52 |
out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
|
53 |
]
|