Skip to content

GitLab

Unverified Commit 8701e864 authored by Yao Matrix's avatar Yao Matrix Committed by GitHub
Browse files

make test_gguf all pass on xpu (#12158) 


Signed-off-by: default avatarYao, Matrix <matrix.yao@intel.com>
parent 58bf2682
Hide whitespace changes
Inline Side-by-side
Showing with 26 additions and 26 deletions
tests/quantization/gguf/test_gguf.py
View file @ 8701e864
...@@ -304,7 +304,7 @@ class FluxGGUFSingleFileTests(GGUFSingleFileTesterMixin, unittest.TestCase): ...@@ -304,7 +304,7 @@ class FluxGGUFSingleFileTests(GGUFSingleFileTesterMixin, unittest.TestCase):
quantization_config=GGUFQuantizationConfig(compute_dtype=torch.bfloat16), quantization_config=GGUFQuantizationConfig(compute_dtype=torch.bfloat16),
config="black-forest-labs/FLUX.1-dev", config="black-forest-labs/FLUX.1-dev",
) )
model.to("cuda") model.to(torch_device)
model(**self.get_dummy_inputs()) model(**self.get_dummy_inputs())
...@@ -360,33 +360,33 @@ class SD35LargeGGUFSingleFileTests(GGUFSingleFileTesterMixin, unittest.TestCase) ...@@ -360,33 +360,33 @@ class SD35LargeGGUFSingleFileTests(GGUFSingleFileTesterMixin, unittest.TestCase)
{ {
("xpu", 3): np.array( ("xpu", 3): np.array(
[ [
0.16210938, 0.1953125,
0.2734375, 0.3125,
0.27734375, 0.31445312,
0.109375, 0.13085938,
0.27148438, 0.30664062,
0.2578125, 0.29296875,
0.1015625, 0.11523438,
0.2578125, 0.2890625,
0.2578125, 0.28320312,
0.14453125, 0.16601562,
0.26953125, 0.3046875,
0.29492188, 0.328125,
0.12890625,
0.28710938,
0.30078125,
0.11132812,
0.27734375,
0.27929688,
0.15625,
0.31054688,
0.296875,
0.15234375,
0.3203125,
0.29492188,
0.140625, 0.140625,
0.31640625,
0.32421875,
0.12304688,
0.3046875, 0.3046875,
0.28515625, 0.3046875,
0.17578125,
0.3359375,
0.3203125,
0.16601562,
0.34375,
0.31640625,
0.15429688,
0.328125,
0.31054688,
] ]
), ),
("cuda", 7): np.array( ("cuda", 7): np.array(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment