Skip to content

Commit c01ea2e

Browse files
committed
align names for t5
1 parent 072e1d9 commit c01ea2e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

optimum/exporters/openvino/model_configs.py

+1
Original file line numberDiff line numberDiff line change
@@ -1850,6 +1850,7 @@ def rename_ambiguous_inputs(self, inputs):
18501850

18511851

18521852
@register_in_tasks_manager("t5-encoder-model", *["feature-extraction"], library_name="diffusers")
1853+
@register_in_tasks_manager("t5-encoder", *["feature-extraction"], library_name="diffusers")
18531854
class T5EncoderOpenVINOConfig(CLIPTextOpenVINOConfig):
18541855
pass
18551856

0 commit comments

Comments
 (0)