Update modeling_quiet.py
Browse files- modeling_quiet.py +1 -1
modeling_quiet.py
CHANGED
|
@@ -1439,7 +1439,7 @@ class QuietForCausalLM(QuietPreTrainedModel, GenerationMixin):
|
|
| 1439 |
|
| 1440 |
from .generate import generate
|
| 1441 |
return generate(self, input_ids=input_ids, attention_mask=attention_mask, max_length=max_length, temperature=temperature, **kwargs)
|
| 1442 |
-
|
| 1443 |
@add_start_docstrings_to_model_forward(QUIET_INPUTS_DOCSTRING)
|
| 1444 |
@replace_return_docstrings(output_type=CausalLMOutputWithPast, config_class=_CONFIG_FOR_DOC)
|
| 1445 |
def forward(
|
|
|
|
| 1439 |
|
| 1440 |
from .generate import generate
|
| 1441 |
return generate(self, input_ids=input_ids, attention_mask=attention_mask, max_length=max_length, temperature=temperature, **kwargs)
|
| 1442 |
+
|
| 1443 |
@add_start_docstrings_to_model_forward(QUIET_INPUTS_DOCSTRING)
|
| 1444 |
@replace_return_docstrings(output_type=CausalLMOutputWithPast, config_class=_CONFIG_FOR_DOC)
|
| 1445 |
def forward(
|