Crystalcareai commited on
Commit
82567c4
·
verified ·
1 Parent(s): d628fc9

Update modeling_quiet.py

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