Pārlūkot izejas kodu

change flags for unit test ci

Alex Cheema 8 mēneši atpakaļ
vecāks
revīzija
41053c55e0
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. 1 1
      .circleci/config.yml

+ 1 - 1
.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