chandan06 commited on
Commit
9210930
·
1 Parent(s): a94b853

Update donut_inference.py

Browse files
Files changed (1) hide show
  1. 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=2,
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,)