rm devices arg

Signed-off-by: Alexandros Koumparoulis <[email protected]>
NVIDIA · Nov 1, 2024 · 777be9c · 777be9c
1 parent ed28860
commit 777be9c
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/.github/workflows/cicd-main.yml b/.github/workflows/cicd-main.yml
@@ -4276,7 +4276,6 @@ jobs:
       SCRIPT: |
 
         python tests/collections/llm/lora_mistralai.py \
-        --devices 2 \
         --max-steps 3 \
         --tp 1 \
         --mbs 1 \
@@ -4291,7 +4290,6 @@ jobs:
       SCRIPT: |
 
         python tests/collections/llm/lora_mistralai.py \
-        --devices 2 \
         --max-steps 3 \
         --tp 1 \
         --mbs 1 \
@@ -4306,7 +4304,6 @@ jobs:
       SCRIPT: |
 
         python tests/collections/llm/lora_mistralai.py \
-        --devices 2 \
         --max-steps 3 \
         --tp 2 \
         --mbs 1 \