Update generate.py
Browse files- generate.py +1 -1
generate.py
CHANGED
|
@@ -3,7 +3,7 @@ import torch
|
|
| 3 |
def generate(model, input_ids, generation_config, left_padding=None, **kwargs):
|
| 4 |
generation_config = generation_config or model.generation_config # default to the model generation config
|
| 5 |
pad_token = kwargs.pop("pad_token", None) or generation_config.pad_token_id or model.config.pad_token_id
|
| 6 |
-
if len(kwargs) > 0:
|
| 7 |
raise ValueError(f"Unused kwargs: {list(kwargs.keys())}")
|
| 8 |
|
| 9 |
cur_length = input_ids.shape[1]
|
|
|
|
| 3 |
def generate(model, input_ids, generation_config, left_padding=None, **kwargs):
|
| 4 |
generation_config = generation_config or model.generation_config # default to the model generation config
|
| 5 |
pad_token = kwargs.pop("pad_token", None) or generation_config.pad_token_id or model.config.pad_token_id
|
| 6 |
+
if len(kwargs) > 0: # Let's catch unexpected kwargs, so that users don't get surprised
|
| 7 |
raise ValueError(f"Unused kwargs: {list(kwargs.keys())}")
|
| 8 |
|
| 9 |
cur_length = input_ids.shape[1]
|