fix order of hidden states in text encoder

huggingface · Oct 31, 2024 · 80f57b5 · 80f57b5
1 parent 5f539d5
commit 80f57b5
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/optimum/intel/openvino/modeling_diffusion.py b/optimum/intel/openvino/modeling_diffusion.py
@@ -995,9 +995,7 @@ def modules(self):
 class OVModelTextEncoder(OVPipelinePart):
     def __init__(self, model: openvino.runtime.Model, parent_pipeline: OVDiffusionPipeline, model_name: str = ""):
         super().__init__(model, parent_pipeline, model_name)
-        self.hidden_states_output_names = sorted(
-            {name for out in self.model.outputs for name in out.names if name.startswith("hidden_states")}
-        )
+        self.hidden_states_output_names = {name for out in self.model.outputs for name in out.names if name.startswith("hidden_states")}
 
     def forward(
         self,