Update modeling_quiet.py
Browse files- modeling_quiet.py +1 -0
modeling_quiet.py
CHANGED
|
@@ -1458,6 +1458,7 @@ class QuietForCausalLM(QuietPreTrainedModel, GenerationMixin):
|
|
| 1458 |
output_attentions: Optional[bool] = None,
|
| 1459 |
output_hidden_states: Optional[bool] = None,
|
| 1460 |
return_dict: Optional[bool] = None,
|
|
|
|
| 1461 |
) -> Union[Tuple, CausalLMOutputWithPast]:
|
| 1462 |
r"""
|
| 1463 |
Args:
|
|
|
|
| 1458 |
output_attentions: Optional[bool] = None,
|
| 1459 |
output_hidden_states: Optional[bool] = None,
|
| 1460 |
return_dict: Optional[bool] = None,
|
| 1461 |
+
temperature_last: Optional[float] = None,
|
| 1462 |
) -> Union[Tuple, CausalLMOutputWithPast]:
|
| 1463 |
r"""
|
| 1464 |
Args:
|