From adb4a16aca1dcb2d013070e34e533feb642d0ca4 Mon Sep 17 00:00:00 2001 From: Alexander Date: Wed, 27 Nov 2024 16:28:30 +0400 Subject: [PATCH] Applied comments --- tests/openvino/test_export.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tests/openvino/test_export.py b/tests/openvino/test_export.py index b2fb9a7b7..80a45cab6 100644 --- 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"])