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