DongHyunKim commited on
Commit
a9bc1da
·
verified ·
1 Parent(s): 427c32f

Update modeling_hyperclovax.py

Browse files
Files changed (1) hide show
  1. modeling_hyperclovax.py +1 -1
modeling_hyperclovax.py CHANGED
@@ -1142,7 +1142,7 @@ class HCXVisionForCausalLM(PreTrainedModel, GenerationMixin):
1142
  pad_token_id=pad_token_id,
1143
  eos_token_id=eos_token_id,
1144
  bad_words_ids=bad_words_ids,
1145
- max_new_tokens=max_length,
1146
  min_length=min_length,
1147
  num_beams=num_beams,
1148
  do_sample=(False if temperature == 0.0 else do_sample), # set do_sample=False if invalid temperature
 
1142
  pad_token_id=pad_token_id,
1143
  eos_token_id=eos_token_id,
1144
  bad_words_ids=bad_words_ids,
1145
+ max_length=max_length,
1146
  min_length=min_length,
1147
  num_beams=num_beams,
1148
  do_sample=(False if temperature == 0.0 else do_sample), # set do_sample=False if invalid temperature