Skip to content

Commit 438fc0e

Browse files
committed
fix awq test
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent 7716cb5 commit 438fc0e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/ipex/test_modeling.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -485,7 +485,7 @@ def test_bnb(self):
485485

486486
@unittest.skipIf(not is_auto_awq_available(), reason="Test requires autoawq")
487487
def test_awq(self):
488-
model_id = "PrunaAI/JackFram-llama-68m-bnb-4bit-smashed"
488+
model_id = "PrunaAI/JackFram-llama-68m-AWQ-4bit-smashed"
489489
set_seed(SEED)
490490
dtype = torch.float16 if IS_XPU_AVAILABLE else torch.float32
491491
# Test model forward do not need cache.

0 commit comments

Comments
 (0)