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

Update modeling_hyperclovax.py

Browse files
Files changed (1) hide show
  1. modeling_hyperclovax.py +1 -0
modeling_hyperclovax.py CHANGED
@@ -1153,6 +1153,7 @@ class HCXVisionForCausalLM(PreTrainedModel, GenerationMixin):
1153
  length_penalty=length_penalty,
1154
  early_stopping=(False if num_beams <= 1 else True), # set early_stopping=False when not beam_search
1155
  use_cache=use_cache,
 
1156
  )
1157
 
1158
  return pred
 
1153
  length_penalty=length_penalty,
1154
  early_stopping=(False if num_beams <= 1 else True), # set early_stopping=False when not beam_search
1155
  use_cache=use_cache,
1156
+ **kwargs,
1157
  )
1158
 
1159
  return pred