Crystalcareai commited on
Commit
5f5f086
·
verified ·
1 Parent(s): 9768b3d

Update modeling_quiet.py

Browse files
Files changed (1) hide show
  1. 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: