@@ -228,7 +228,7 @@ def test_mask(self):
228
228
expected_slice = np .array ([0 ] * 9 )
229
229
max_diff = np .abs (mask_slice .flatten () - expected_slice ).max ()
230
230
self .assertLessEqual (max_diff , 1e-3 )
231
- self .assertEqual (mask [0 , - 3 , - 4 ], 1 )
231
+ self .assertEqual (mask [0 , - 3 , - 4 ], 0 )
232
232
233
233
def test_inversion (self ):
234
234
device = "cpu"
@@ -244,7 +244,7 @@ def test_inversion(self):
244
244
245
245
self .assertEqual (image .shape , (2 , 32 , 32 , 3 ))
246
246
expected_slice = np .array (
247
- [0.5588859 , 0.535619 , 0.52224344 , 0.55604255 , 0.48608556 , 0.51105076 , 0.50301707 , 0.44348782 , 0.48488846 ],
247
+ [0.5150 , 0.5134 , 0.5043 , 0.5376 , 0.4694 , 0.51050 , 0.5015 , 0.4407 , 0.4799 ],
248
248
)
249
249
max_diff = np .abs (image_slice .flatten () - expected_slice ).max ()
250
250
self .assertLessEqual (max_diff , 1e-3 )
@@ -305,6 +305,8 @@ def test_stable_diffusion_diffedit_full(self):
305
305
).images [0 ]
306
306
307
307
expected_image = np .array (
308
- load_image ("https://raw.githubusercontent.com/Xiang-cd/DiffEdit-stable-diffusion/main/assets/target.png" )
308
+ load_image (
309
+ "https://raw.githubusercontent.com/Xiang-cd/DiffEdit-stable-diffusion/main/assets/target.png"
310
+ ).resize ((768 , 768 ))
309
311
)
310
312
assert np .abs ((expected_image - image ).max ()) < 1e-1
0 commit comments