KeyError: 'phi model type is not supported yet in NormalizedConfig.

[eaidova]

System Info

optimum 1.16.1
platform ubuntu-20
python python3.8

Who can help?

No response

Information

• The official example scripts
• My own modified scripts

Tasks

• An officially supported task in the examples folder (such as GLUE/SQuAD, ...)
• My own task or dataset (give details below)

Reproduction (minimal, reproducible, runnable)

due to missing NormalizedConfig registration for phi model, I can not load model using optimum.intel.openvino.OVModelForCausalLM:

from optimum.intel.openvino import OVModelForCausalLM

model_id = "susnato/phi-1_5_dev"
ov_model = OVModelForCausalLM.from_pretrained(model_id, export=True, compile=False)

as a workaround, if I add before creating the model these lines:

from optimum.utils import NormalizedTextConfig, NormalizedConfigManager

NormalizedConfigManager._conf['phi'] = NormalizedTextConfig

it can be successfully loaded.
I suppose that these lines missed in
https://github.com/huggingface/optimum/blob/main/optimum/utils/normalized_config.py#L210
as it defined in OnnxExportConfig https://github.com/huggingface/optimum/blob/main/optimum/exporters/onnx/model_configs.py#L253 https://github.com/huggingface/optimum/blob/main/optimum/exporters/tasks.py#L816

Expected behavior

model converted and loaded without errors

[fxmarty]

Hi @eaidova, this should be fixed with #1625