Update generate.py
Browse files- generate.py +1 -1
generate.py
CHANGED
|
@@ -164,7 +164,7 @@ def generate(
|
|
| 164 |
|
| 165 |
# Validate stopping criteria
|
| 166 |
stopping_criteria = validate_stopping_criteria(
|
| 167 |
-
max_length, max_new_tokens=None, min_length=min_length, stopping_criteria=self._get_stopping_criteria()
|
| 168 |
)
|
| 169 |
# Generate using the custom generate function
|
| 170 |
generated_token_ids = custom_generate(
|
|
|
|
| 164 |
|
| 165 |
# Validate stopping criteria
|
| 166 |
stopping_criteria = validate_stopping_criteria(
|
| 167 |
+
max_length, max_new_tokens=None, min_length=min_length, stopping_criteria=self._get_stopping_criteria(generation_config=self.config, stopping_criteria=StoppingCriteriaList())
|
| 168 |
)
|
| 169 |
# Generate using the custom generate function
|
| 170 |
generated_token_ids = custom_generate(
|