Update modeling_quiet.py
Browse files- modeling_quiet.py +1 -0
modeling_quiet.py
CHANGED
|
@@ -1439,6 +1439,7 @@ class QuietForCausalLM(QuietPreTrainedModel, GenerationMixin):
|
|
| 1439 |
output_attentions: Optional[bool] = None,
|
| 1440 |
output_hidden_states: Optional[bool] = None,
|
| 1441 |
return_dict: Optional[bool] = None,
|
|
|
|
| 1442 |
) -> Union[Tuple, CausalLMOutputWithPast]:
|
| 1443 |
r"""
|
| 1444 |
Args:
|
|
|
|
| 1439 |
output_attentions: Optional[bool] = None,
|
| 1440 |
output_hidden_states: Optional[bool] = None,
|
| 1441 |
return_dict: Optional[bool] = None,
|
| 1442 |
+
streamer: Optional[TextStreamer] = None,
|
| 1443 |
) -> Union[Tuple, CausalLMOutputWithPast]:
|
| 1444 |
r"""
|
| 1445 |
Args:
|