Update donut_inference.py
Browse files- donut_inference.py +1 -1
donut_inference.py
CHANGED
|
@@ -35,7 +35,7 @@ def inference(image):
|
|
| 35 |
pad_token_id=processor.tokenizer.pad_token_id,
|
| 36 |
eos_token_id=processor.tokenizer.eos_token_id,
|
| 37 |
use_cache=True,
|
| 38 |
-
num_beams=
|
| 39 |
bad_words_ids=[[processor.tokenizer.unk_token_id]],
|
| 40 |
return_dict_in_generate=True,
|
| 41 |
output_scores=True,)
|
|
|
|
| 35 |
pad_token_id=processor.tokenizer.pad_token_id,
|
| 36 |
eos_token_id=processor.tokenizer.eos_token_id,
|
| 37 |
use_cache=True,
|
| 38 |
+
num_beams=1,
|
| 39 |
bad_words_ids=[[processor.tokenizer.unk_token_id]],
|
| 40 |
return_dict_in_generate=True,
|
| 41 |
output_scores=True,)
|