oddadmix commited on
Commit
2116e87
·
verified ·
1 Parent(s): 77a9339

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
  ]