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
|