w3robotics
commited on
Update app.py
Browse files
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=
|
79 |
-
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,
|