Update modeling_quiet.py
Browse files- modeling_quiet.py +2 -2
modeling_quiet.py
CHANGED
|
@@ -1464,8 +1464,8 @@ class QuietForCausalLM(QuietPreTrainedModel, GenerationMixin):
|
|
| 1464 |
stopping_criteria: Optional[StoppingCriteriaList] = None,
|
| 1465 |
**model_kwargs,
|
| 1466 |
):
|
| 1467 |
-
# Validate stopping criteria
|
| 1468 |
-
stopping_criteria = validate_stopping_criteria(stopping_criteria, max_length)
|
| 1469 |
|
| 1470 |
# Set default values for attention_mask and max_new_tokens
|
| 1471 |
if "attention_mask" not in model_kwargs:
|
|
|
|
| 1464 |
stopping_criteria: Optional[StoppingCriteriaList] = None,
|
| 1465 |
**model_kwargs,
|
| 1466 |
):
|
| 1467 |
+
# # Validate stopping criteria
|
| 1468 |
+
# stopping_criteria = validate_stopping_criteria(stopping_criteria, max_length)
|
| 1469 |
|
| 1470 |
# Set default values for attention_mask and max_new_tokens
|
| 1471 |
if "attention_mask" not in model_kwargs:
|