Update modeling_hyperclovax.py
Browse files- 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
|