fix params

Signed-off-by: Alexandros Koumparoulis <[email protected]>
NVIDIA · Oct 31, 2024 · d2fb454 · d2fb454
1 parent 83efb9c
commit d2fb454
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.github/workflows/cicd-main.yml b/.github/workflows/cicd-main.yml
@@ -4331,7 +4331,7 @@ jobs:
 
         python tests/collections/llm/lora_mistralai.py \
         --devices 2 \
-        --max_steps 3 \
+        --max-steps 3 \
         --tp 1 \
         --mbs 1 \
         --model mixtral
@@ -4346,7 +4346,7 @@ jobs:
 
         python tests/collections/llm/lora_mistralai.py \
         --devices 2 \
-        --max_steps 3 \
+        --max-steps 3 \
         --tp 1 \
         --mbs 1 \
         --model mistral
@@ -4361,7 +4361,7 @@ jobs:
 
         python tests/collections/llm/lora_mistralai.py \
         --devices 2 \
-        --max_steps 3 \
+        --max-steps 3 \
         --tp 2 \
         --mbs 1 \
         --model mistral