align names for t5

huggingface · Dec 23, 2024 · c01ea2e · c01ea2e
1 parent 072e1d9
commit c01ea2e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/optimum/exporters/openvino/model_configs.py b/optimum/exporters/openvino/model_configs.py
@@ -1850,6 +1850,7 @@ def rename_ambiguous_inputs(self, inputs):
 
 
 @register_in_tasks_manager("t5-encoder-model", *["feature-extraction"], library_name="diffusers")
+@register_in_tasks_manager("t5-encoder", *["feature-extraction"], library_name="diffusers")
 class T5EncoderOpenVINOConfig(CLIPTextOpenVINOConfig):
     pass