fix backend not ipex (#930)

* set env to DEBUG * add logs * fix backend not ipex
ModelCloud · Dec 19, 2024 · 1e1e3b6 · 1e1e3b6
1 parent fa7929b
commit 1e1e3b6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml
@@ -38,6 +38,7 @@ on:
         default: '10'
 
 env:
+  DEBUG: 1
   CUDA_DEVICE_ORDER: PCI_BUS_ID
   CUDA_VISIBLE_DEVICES: 0
   TORCH_CUDA_ARCH_LIST: '8.9'

diff --git a/tests/test_ipex_xpu.py b/tests/test_ipex_xpu.py
@@ -25,7 +25,7 @@ def test(self):
         )
         tokenizer = self.load_tokenizer(self.NATIVE_MODEL_ID)
         calibration_dataset = self.load_dataset(tokenizer)
-        origin_model.quantize(calibration_dataset)
+        origin_model.quantize(calibration_dataset, backend=BACKEND.IPEX)
         with tempfile.TemporaryDirectory() as tmpdir:
           origin_model.save(tmpdir)