w3robotics commited on
Commit
90d0fc5
·
verified ·
1 Parent(s): 29d33c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -75,8 +75,8 @@ if file_name is not None:
75
  pixel_values.to(device),
76
  decoder_input_ids=decoder_input_ids.to(device),
77
  max_length=model_ext.decoder.config.max_position_embeddings,
78
- pad_token_id=processor.tokenizer.pad_token_id,
79
- eos_token_id=processor.tokenizer.eos_token_id,
80
  use_cache=True,
81
  bad_words_ids=[[processor_ext.tokenizer.unk_token_id]],
82
  return_dict_in_generate=True,
 
75
  pixel_values.to(device),
76
  decoder_input_ids=decoder_input_ids.to(device),
77
  max_length=model_ext.decoder.config.max_position_embeddings,
78
+ pad_token_id=processor_ext.tokenizer.pad_token_id,
79
+ eos_token_id=processor_ext.tokenizer.eos_token_id,
80
  use_cache=True,
81
  bad_words_ids=[[processor_ext.tokenizer.unk_token_id]],
82
  return_dict_in_generate=True,