@@ -192,112 +192,16 @@ def test_sd3_img2img_inference(self):
192192        inputs  =  self .get_inputs (torch_device )
193193        image  =  pipe (** inputs ).images [0 ]
194194        image_slice  =  image [0 , :10 , :10 ]
195+ 
196+         # fmt: off 
195197        expected_slices  =  Expectations (
196198            {
197-                 ("xpu" , 3 ): np .array (
198-                     [
199-                         0.5117 ,
200-                         0.4421 ,
201-                         0.3852 ,
202-                         0.5044 ,
203-                         0.4219 ,
204-                         0.3262 ,
205-                         0.5024 ,
206-                         0.4329 ,
207-                         0.3276 ,
208-                         0.4978 ,
209-                         0.4412 ,
210-                         0.3355 ,
211-                         0.4983 ,
212-                         0.4338 ,
213-                         0.3279 ,
214-                         0.4893 ,
215-                         0.4241 ,
216-                         0.3129 ,
217-                         0.4875 ,
218-                         0.4253 ,
219-                         0.3030 ,
220-                         0.4961 ,
221-                         0.4267 ,
222-                         0.2988 ,
223-                         0.5029 ,
224-                         0.4255 ,
225-                         0.3054 ,
226-                         0.5132 ,
227-                         0.4248 ,
228-                         0.3222 ,
229-                     ]
230-                 ),
231-                 ("cuda" , 7 ): np .array (
232-                     [
233-                         0.5435 ,
234-                         0.4673 ,
235-                         0.5732 ,
236-                         0.4438 ,
237-                         0.3557 ,
238-                         0.4912 ,
239-                         0.4331 ,
240-                         0.3491 ,
241-                         0.4915 ,
242-                         0.4287 ,
243-                         0.347 ,
244-                         0.4849 ,
245-                         0.4355 ,
246-                         0.3469 ,
247-                         0.4871 ,
248-                         0.4431 ,
249-                         0.3538 ,
250-                         0.4912 ,
251-                         0.4521 ,
252-                         0.3643 ,
253-                         0.5059 ,
254-                         0.4587 ,
255-                         0.373 ,
256-                         0.5166 ,
257-                         0.4685 ,
258-                         0.3845 ,
259-                         0.5264 ,
260-                         0.4746 ,
261-                         0.3914 ,
262-                         0.5342 ,
263-                     ]
264-                 ),
265-                 ("cuda" , 8 ): np .array (
266-                     [
267-                         0.5146 ,
268-                         0.4385 ,
269-                         0.3826 ,
270-                         0.5098 ,
271-                         0.4150 ,
272-                         0.3218 ,
273-                         0.5142 ,
274-                         0.4312 ,
275-                         0.3298 ,
276-                         0.5127 ,
277-                         0.4431 ,
278-                         0.3411 ,
279-                         0.5171 ,
280-                         0.4424 ,
281-                         0.3374 ,
282-                         0.5088 ,
283-                         0.4348 ,
284-                         0.3242 ,
285-                         0.5073 ,
286-                         0.4380 ,
287-                         0.3174 ,
288-                         0.5132 ,
289-                         0.4397 ,
290-                         0.3115 ,
291-                         0.5132 ,
292-                         0.4343 ,
293-                         0.3118 ,
294-                         0.5219 ,
295-                         0.4328 ,
296-                         0.3256 ,
297-                     ]
298-                 ),
199+                 ("xpu" , 3 ): np .array ([0.5117 , 0.4421 , 0.3852 , 0.5044 , 0.4219 , 0.3262 , 0.5024 , 0.4329 , 0.3276 , 0.4978 , 0.4412 , 0.3355 , 0.4983 , 0.4338 , 0.3279 , 0.4893 , 0.4241 , 0.3129 , 0.4875 , 0.4253 , 0.3030 , 0.4961 , 0.4267 , 0.2988 , 0.5029 , 0.4255 , 0.3054 , 0.5132 , 0.4248 , 0.3222 ]),
200+                 ("cuda" , 7 ): np .array ([0.5435 , 0.4673 , 0.5732 , 0.4438 , 0.3557 , 0.4912 , 0.4331 , 0.3491 , 0.4915 , 0.4287 , 0.347 , 0.4849 , 0.4355 , 0.3469 , 0.4871 , 0.4431 , 0.3538 , 0.4912 , 0.4521 , 0.3643 , 0.5059 , 0.4587 , 0.373 , 0.5166 , 0.4685 , 0.3845 , 0.5264 , 0.4746 , 0.3914 , 0.5342 ]),
201+                 ("cuda" , 8 ): np .array ([0.5146 , 0.4385 , 0.3826 , 0.5098 , 0.4150 , 0.3218 , 0.5142 , 0.4312 , 0.3298 , 0.5127 , 0.4431 , 0.3411 , 0.5171 , 0.4424 , 0.3374 , 0.5088 , 0.4348 , 0.3242 , 0.5073 , 0.4380 , 0.3174 , 0.5132 , 0.4397 , 0.3115 , 0.5132 , 0.4343 , 0.3118 , 0.5219 , 0.4328 , 0.3256 ]),
299202            }
300203        )
204+         # fmt: on 
301205
302206        expected_slice  =  expected_slices .get_expectation ()
303207
0 commit comments