@@ -436,6 +436,9 @@ def test_unidiffuser_img2text_multiple_prompts_with_latents(self):
436436
437437 assert len (text ) == 3
438438
439+ def test_inference_batch_single_identical (self ):
440+ super ().test_inference_batch_single_identical (expected_max_diff = 2e-4 )
441+
439442 @require_torch_gpu
440443 def test_unidiffuser_default_joint_v1_cuda_fp16 (self ):
441444 device = "cuda"
@@ -583,7 +586,7 @@ def test_unidiffuser_default_joint_v1(self):
583586 expected_img_slice = np .array ([0.2402 , 0.2375 , 0.2285 , 0.2378 , 0.2407 , 0.2263 , 0.2354 , 0.2307 , 0.2520 ])
584587 assert np .abs (image_slice .flatten () - expected_img_slice ).max () < 1e-1
585588
586- expected_text_prefix = "A living room"
589+ expected_text_prefix = "a living room"
587590 assert text [0 ][: len (expected_text_prefix )] == expected_text_prefix
588591
589592 def test_unidiffuser_default_text2img_v1 (self ):
@@ -634,9 +637,9 @@ def test_unidiffuser_default_joint_v1_fp16(self):
634637
635638 image_slice = image [0 , - 3 :, - 3 :, - 1 ]
636639 expected_img_slice = np .array ([0.2402 , 0.2375 , 0.2285 , 0.2378 , 0.2407 , 0.2263 , 0.2354 , 0.2307 , 0.2520 ])
637- assert np .abs (image_slice .flatten () - expected_img_slice ).max () < 1e -1
640+ assert np .abs (image_slice .flatten () - expected_img_slice ).max () < 2e -1
638641
639- expected_text_prefix = "A living room"
642+ expected_text_prefix = "a living room"
640643 assert text [0 ][: len (expected_text_prefix )] == expected_text_prefix
641644
642645 def test_unidiffuser_default_text2img_v1_fp16 (self ):
0 commit comments