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):
304304 quantization_config = GGUFQuantizationConfig (compute_dtype = torch .bfloat16 ),
305305 config = "black-forest-labs/FLUX.1-dev" ,
306306 )
307- model .to ("cuda" )
307+ model .to (torch_device )
308308 model (** self .get_dummy_inputs ())
309309
310310
@@ -360,33 +360,33 @@ def test_pipeline_inference(self):
360360 {
361361 ("xpu" , 3 ): np .array (
362362 [
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 ,
387375 0.140625 ,
376+ 0.31640625 ,
377+ 0.32421875 ,
378+ 0.12304688 ,
388379 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 ,
390390 ]
391391 ),
392392 ("cuda" , 7 ): np .array (
You can’t perform that action at this time.
0 commit comments