Skip to content

Commit 8701e86

Browse files
authored
make test_gguf all pass on xpu (#12158)
Signed-off-by: Yao, Matrix <[email protected]>
1 parent 58bf268 commit 8701e86

File tree

1 file changed

+26
-26
lines changed

1 file changed

+26
-26
lines changed

tests/quantization/gguf/test_gguf.py

Lines changed: 26 additions & 26 deletions
Original file line numberDiff line numberDiff 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(

0 commit comments

Comments
 (0)