Skip to content

Commit 4db4db5

Browse files
committed
fix tests
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent 733bbc4 commit 4db4db5

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
@@ -374,7 +374,7 @@ def test_ipex_beam_search(self, test_name, model_arch, use_cache):
374374
model_id, use_cache=use_cache, torch_dtype=dtype, device_map=DEVICE
375375
)
376376
# It will be removed when torch 2.6 released
377-
if model_arch == "opt" and not use_cache and model.compiled and is_torch_version("<", "2.6.0"):
377+
if model_arch == "opt" and not use_cache and model.compile and is_torch_version("<", "2.6.0"):
378378
return
379379
if use_cache and model_arch in self.IPEX_PATCHED_SUPPORTED_ARCHITECTURES:
380380
self.assertTrue(model.add_patch)

0 commit comments

Comments
 (0)