diff --git a/examples/models/llama/config/test_llm_config.py b/examples/models/llama/config/test_llm_config.py index 0853e9dbbd8..2e8e0591184 100644 --- a/examples/models/llama/config/test_llm_config.py +++ b/examples/models/llama/config/test_llm_config.py @@ -94,7 +94,7 @@ def test_valid_llm_config(self): backend=BackendConfig( xnnpack=XNNPackConfig(enabled=False), coreml=CoreMLConfig( - enabled=True, ios=17, compute_units=CoreMLComputeUnit.ALL + enabled=True, ios=17, compute_units=CoreMLComputeUnit.cpu_only ), ), )