Skip to content

Commit 28bb4f0

Browse files
Fix model field names for OVBaseModelForSeq2SeqLM
1 parent 88e16b5 commit 28bb4f0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

optimum/intel/openvino/modeling_base_seq2seq.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -108,9 +108,9 @@ def __init__(
108108

109109
@property
110110
def _ov_submodel_names(self) -> List[str]:
111-
submodel_names = ["encoder", "decoder"]
111+
submodel_names = ["encoder_model", "decoder_model"]
112112
if self.decoder_with_past_model is not None:
113-
submodel_names.append("decoder_with_past")
113+
submodel_names.append("decoder_with_past_model")
114114
return submodel_names
115115

116116
def _save_pretrained(self, save_directory: Union[str, Path]):

0 commit comments

Comments
 (0)