Crystalcareai
commited on
Update modeling_quiet.py
Browse files- modeling_quiet.py +1 -0
modeling_quiet.py
CHANGED
@@ -1176,6 +1176,7 @@ class QuietForCausalLM(QuietPreTrainedModel, GenerationMixin):
|
|
1176 |
prompt_lookup_num_tokens: Optional[int] = None,
|
1177 |
epsilon_cutoff: Optional[float] = None,
|
1178 |
eta_cutoff: Optional[float] = None,
|
|
|
1179 |
suppress_tokens: Optional[List[int]] = None,
|
1180 |
synced_gpus: Optional[bool] = None,
|
1181 |
eos_token_id: Optional[List[int]] = None,
|
|
|
1176 |
prompt_lookup_num_tokens: Optional[int] = None,
|
1177 |
epsilon_cutoff: Optional[float] = None,
|
1178 |
eta_cutoff: Optional[float] = None,
|
1179 |
+
max_length: Optional[int] = None,
|
1180 |
suppress_tokens: Optional[List[int]] = None,
|
1181 |
synced_gpus: Optional[bool] = None,
|
1182 |
eos_token_id: Optional[List[int]] = None,
|