Crystalcareai
commited on
Update modeling_quiet.py
Browse files- modeling_quiet.py +1 -1
modeling_quiet.py
CHANGED
@@ -1424,7 +1424,7 @@ class QuietForCausalLM(QuietPreTrainedModel, GenerationMixin):
|
|
1424 |
|
1425 |
def generate(self, input_ids, attention_mask=None, max_length=None, temperature=1.0, **kwargs):
|
1426 |
from .generate import generate
|
1427 |
-
return generate(self, input_ids, attention_mask
|
1428 |
|
1429 |
@add_start_docstrings_to_model_forward(QUIET_INPUTS_DOCSTRING)
|
1430 |
@replace_return_docstrings(output_type=CausalLMOutputWithPast, config_class=_CONFIG_FOR_DOC)
|
|
|
1424 |
|
1425 |
def generate(self, input_ids, attention_mask=None, max_length=None, temperature=1.0, **kwargs):
|
1426 |
from .generate import generate
|
1427 |
+
return generate(self, input_ids, attention_mask, max_length, temperature, **kwargs)
|
1428 |
|
1429 |
@add_start_docstrings_to_model_forward(QUIET_INPUTS_DOCSTRING)
|
1430 |
@replace_return_docstrings(output_type=CausalLMOutputWithPast, config_class=_CONFIG_FOR_DOC)
|