File tree Expand file tree Collapse file tree 1 file changed +26
-26
lines changed Expand file tree Collapse file tree 1 file changed +26
-26
lines changed Original file line number Diff line number Diff line change @@ -304,7 +304,7 @@ def test_loading_gguf_diffusers_format(self):
304
304
quantization_config = GGUFQuantizationConfig (compute_dtype = torch .bfloat16 ),
305
305
config = "black-forest-labs/FLUX.1-dev" ,
306
306
)
307
- model .to ("cuda" )
307
+ model .to (torch_device )
308
308
model (** self .get_dummy_inputs ())
309
309
310
310
@@ -360,33 +360,33 @@ def test_pipeline_inference(self):
360
360
{
361
361
("xpu" , 3 ): np .array (
362
362
[
363
- 0.16210938 ,
364
- 0.2734375 ,
365
- 0.27734375 ,
366
- 0.109375 ,
367
- 0.27148438 ,
368
- 0.2578125 ,
369
- 0.1015625 ,
370
- 0.2578125 ,
371
- 0.2578125 ,
372
- 0.14453125 ,
373
- 0.26953125 ,
374
- 0.29492188 ,
375
- 0.12890625 ,
376
- 0.28710938 ,
377
- 0.30078125 ,
378
- 0.11132812 ,
379
- 0.27734375 ,
380
- 0.27929688 ,
381
- 0.15625 ,
382
- 0.31054688 ,
383
- 0.296875 ,
384
- 0.15234375 ,
385
- 0.3203125 ,
386
- 0.29492188 ,
363
+ 0.1953125 ,
364
+ 0.3125 ,
365
+ 0.31445312 ,
366
+ 0.13085938 ,
367
+ 0.30664062 ,
368
+ 0.29296875 ,
369
+ 0.11523438 ,
370
+ 0.2890625 ,
371
+ 0.28320312 ,
372
+ 0.16601562 ,
373
+ 0.3046875 ,
374
+ 0.328125 ,
387
375
0.140625 ,
376
+ 0.31640625 ,
377
+ 0.32421875 ,
378
+ 0.12304688 ,
388
379
0.3046875 ,
389
- 0.28515625 ,
380
+ 0.3046875 ,
381
+ 0.17578125 ,
382
+ 0.3359375 ,
383
+ 0.3203125 ,
384
+ 0.16601562 ,
385
+ 0.34375 ,
386
+ 0.31640625 ,
387
+ 0.15429688 ,
388
+ 0.328125 ,
389
+ 0.31054688 ,
390
390
]
391
391
),
392
392
("cuda" , 7 ): np .array (
You can’t perform that action at this time.
0 commit comments