@@ -144,6 +144,17 @@ def test_compare_to_diffusers_pipeline(self, model_arch: str):
144
144
145
145
np .testing .assert_allclose (ov_output , diffusers_output , atol = 6e-3 , rtol = 1e-2 )
146
146
147
+ # test on inputs nondivisible on 64
148
+ height , width , batch_size = 96 , 96 , 1
149
+
150
+ for output_type in ["latent" , "np" , "pt" ]:
151
+ inputs ["output_type" ] = output_type
152
+
153
+ ov_output = ov_pipeline (** inputs , generator = get_generator ("pt" , SEED )).images
154
+ diffusers_output = diffusers_pipeline (** inputs , generator = get_generator ("pt" , SEED )).images
155
+
156
+ np .testing .assert_allclose (ov_output , diffusers_output , atol = 6e-3 , rtol = 1e-2 )
157
+
147
158
@parameterized .expand (CALLBACK_SUPPORT_ARCHITECTURES )
148
159
@require_diffusers
149
160
def test_callback (self , model_arch : str ):
@@ -541,6 +552,20 @@ def test_compare_to_diffusers_pipeline(self, model_arch: str):
541
552
542
553
np .testing .assert_allclose (ov_output , diffusers_output , atol = 6e-3 , rtol = 1e-2 )
543
554
555
+ # test generation when input resolution nondevisible on 64
556
+ height , width , batch_size = 96 , 96 , 1
557
+
558
+ inputs = self .generate_inputs (height = height , width = width , batch_size = batch_size , model_type = model_arch )
559
+
560
+ for output_type in ["latent" , "np" , "pt" ]:
561
+ print (output_type )
562
+ inputs ["output_type" ] = output_type
563
+
564
+ ov_output = ov_pipeline (** inputs , generator = get_generator ("pt" , SEED )).images
565
+ diffusers_output = diffusers_pipeline (** inputs , generator = get_generator ("pt" , SEED )).images
566
+
567
+ np .testing .assert_allclose (ov_output , diffusers_output , atol = 6e-3 , rtol = 1e-2 )
568
+
544
569
@parameterized .expand (SUPPORTED_ARCHITECTURES )
545
570
@require_diffusers
546
571
def test_image_reproducibility (self , model_arch : str ):
@@ -777,6 +802,18 @@ def test_compare_to_diffusers_pipeline(self, model_arch: str):
777
802
778
803
np .testing .assert_allclose (ov_output , diffusers_output , atol = 6e-3 , rtol = 1e-2 )
779
804
805
+ # test generation when input resolution nondevisible on 64
806
+ height , width , batch_size = 96 , 96 , 1
807
+ inputs = self .generate_inputs (height = height , width = width , batch_size = batch_size )
808
+
809
+ for output_type in ["latent" , "np" , "pt" ]:
810
+ inputs ["output_type" ] = output_type
811
+
812
+ ov_output = ov_pipeline (** inputs , generator = get_generator ("pt" , SEED )).images
813
+ diffusers_output = diffusers_pipeline (** inputs , generator = get_generator ("pt" , SEED )).images
814
+
815
+ np .testing .assert_allclose (ov_output , diffusers_output , atol = 6e-3 , rtol = 1e-2 )
816
+
780
817
@parameterized .expand (SUPPORTED_ARCHITECTURES )
781
818
@require_diffusers
782
819
def test_image_reproducibility (self , model_arch : str ):
0 commit comments