Applied comments

huggingface · Nov 27, 2024 · adb4a16 · adb4a16
1 parent 66d5d9c
commit adb4a16
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/openvino/test_export.py b/tests/openvino/test_export.py
@@ -144,6 +144,14 @@ def _openvino_export(
                         self.assertTrue(
                             ov_model.text_encoder.model.has_rt_info(["runtime_options", "ACTIVATIONS_SCALE_FACTOR"])
                         )
+                    if hasattr(ov_model, "text_encoder_2") and ov_model.text_encoder_2:
+                        self.assertTrue(
+                            ov_model.text_encoder_2.model.has_rt_info(["runtime_options", "ACTIVATIONS_SCALE_FACTOR"])
+                        )
+                    if hasattr(ov_model, "text_encoder_3") and ov_model.text_encoder_3:
+                        self.assertTrue(
+                            ov_model.text_encoder_3.model.has_rt_info(["runtime_options", "ACTIVATIONS_SCALE_FACTOR"])
+                        )
                     if hasattr(ov_model, "unet") and ov_model.unet:
                         self.assertTrue(
                             ov_model.unet.model.has_rt_info(["runtime_options", "ACTIVATIONS_SCALE_FACTOR"])