We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 88e16b5 commit 1ec3f38Copy full SHA for 1ec3f38
optimum/intel/openvino/modeling_base_seq2seq.py
@@ -108,9 +108,9 @@ def __init__(
108
109
@property
110
def _ov_submodel_names(self) -> List[str]:
111
- submodel_names = ["encoder", "decoder"]
+ submodel_names = ["encoder_model", "decoder_model"]
112
if self.decoder_with_past_model is not None:
113
- submodel_names.append("decoder_with_past")
+ submodel_names.append("decoder_with_past_model")
114
return submodel_names
115
116
def _save_pretrained(self, save_directory: Union[str, Path]):
0 commit comments