Skip to content

Commit 860bdf8

Browse files
authored
fix updated broken checking nondefault parameters (#1038)
1 parent da5903d commit 860bdf8

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

optimum/exporters/openvino/convert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -692,7 +692,7 @@ def export_from_model(
692692
# some model configs may have issues with loading without parameters initialization
693693
try:
694694
misplaced_generation_parameters = model.config._get_non_default_generation_parameters()
695-
except KeyError:
695+
except (KeyError, TypeError):
696696
misplaced_generation_parameters = {}
697697
if isinstance(model, GenerationMixin) and len(misplaced_generation_parameters) > 0:
698698
logger.warning(

optimum/intel/openvino/modeling_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ def __init__(
139139
# some model configs may have issues with loading without parameters initialization
140140
try:
141141
misplaced_generation_parameters = self.config._get_non_default_generation_parameters()
142-
except KeyError:
142+
except (KeyError, TypeError):
143143
misplaced_generation_parameters = {}
144144
if len(misplaced_generation_parameters) > 0:
145145
logger.warning(

optimum/intel/openvino/modeling_base_seq2seq.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def __init__(
8787
# some model configs may have issues with loading without parameters initialization
8888
try:
8989
misplaced_generation_parameters = self.config._get_non_default_generation_parameters()
90-
except KeyError:
90+
except (KeyError, TypeError):
9191
misplaced_generation_parameters = {}
9292
if len(misplaced_generation_parameters) > 0:
9393
logger.warning(

0 commit comments

Comments
 (0)