Skip to content

Commit 6d879da

Browse files
Update optimum/exporters/openvino/model_configs.py
Co-authored-by: Ella Charlaix <[email protected]>
1 parent 2a9bd5d commit 6d879da

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

optimum/exporters/openvino/model_configs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4271,8 +4271,8 @@ def generate_dummy_inputs(self, framework: str = "pt", **kwargs):
42714271
)
42724272
class HunyuanOpenVINOConfig(TextDecoderWithPositionIdsOnnxConfig):
42734273
MIN_TRANSFORMERS_VERSION = "4.56.0"
4274-
DUMMY_INPUT_GENERATOR_CLASSES = (DummyTextInputGenerator, HunyuanDummyPastKeyValuesGenerator)
4275-
DUMMY_PKV_GENERATOR_CLASS = HunyuanDummyPastKeyValuesGenerator
4274+
DUMMY_INPUT_GENERATOR_CLASSES = (DummyTextInputGenerator, MistralDummyPastKeyValuesGenerator)
4275+
DUMMY_PKV_GENERATOR_CLASS = MistralDummyPastKeyValuesGenerator
42764276
NORMALIZED_CONFIG_CLASS = NormalizedTextConfig
42774277
_MODEL_PATCHER = OVDecoderModelPatcher
42784278

0 commit comments

Comments
 (0)