change flags for unit test ci

41053c55e051b2620694e03acb5e44946b011219 · 2024-09-24 21:48:42 +0100 · Alex Cheema

Files touched

M .circleci/config.yml

Diff

commit 41053c55e051b2620694e03acb5e44946b011219
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Tue Sep 24 21:48:42 2024 +0100

    change flags for unit test ci
---
 .circleci/config.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.circleci/config.yml b/.circleci/config.yml
index fcf4c324..6de2cac6 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -113,7 +113,7 @@ jobs:
             source env/bin/activate
             # set TEMPERATURE to 0 for deterministic sampling
             echo "Running inference engine tests..."
-            LOW_MEMORY=1 DEVICE=gpu METAL_DEVICE_WRAPPER_TYPE=1 METAL_DEBUG_ERROR_MODE=0 METAL_XCODE=1 TEMPERATURE=0 python3 -m exo.inference.test_inference_engine
+            METAL_DEVICE_WRAPPER_TYPE=1 METAL_DEBUG_ERROR_MODE=0 METAL_XCODE=1 TEMPERATURE=0 python3 -m exo.inference.test_inference_engine
             echo "Running tokenizer tests..."
             python3 ./test/test_tokenizers.py

← 10812c45 test ci · back to Exo · bump up tinygrad to 232edcfd4f8b388807c64fb1817a7668ce27cbad 04df31b1 →