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