Move CI to mi300x-3

nod-ai · Oct 25, 2024 · 006c5d4 · 006c5d4
1 parent 3a91051
commit 006c5d4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/ci_eval.yaml b/.github/workflows/ci_eval.yaml
@@ -22,7 +22,7 @@ jobs:
     strategy:
       matrix:
         version: [3.11]
-        runs-on: [llama-mi300]
+        runs-on: [llama-mi300x-3]
       fail-fast: false
     runs-on: ${{matrix.runs-on}}
     defaults:

diff --git a/sharktank/conftest.py b/sharktank/conftest.py
@@ -76,7 +76,7 @@ def pytest_addoption(parser):
         "--llama3-8b-tokenizer-path",
         type=Path,
         action="store",
-        default="/data/extra/models/llama3.1_8B/tokenizer_config.json",
+        default="/data/llama-3.1/8b/tokenizer_config.json",
         help="Llama3.1 8b tokenizer path, defaults to 30F CI system path",
     )
 
@@ -92,7 +92,7 @@ def pytest_addoption(parser):
         "--llama3-8b-f16-model-path",
         type=Path,
         action="store",
-        default="/data/extra/models/llama3.1_8B/llama8b_f16.irpa",
+        default="/data/llama-3.1/8b/llama8b_f16.irpa",
         help="Llama3.1 8b model path, defaults to 30F CI system path",
     )
 
@@ -132,7 +132,7 @@ def pytest_addoption(parser):
         "--llama3-405b-tokenizer-path",
         type=Path,
         action="store",
-        default="/data/extra/models/llama3.1_405B/tokenizer_config.json",
+        default="/data/llama-3.1/405b/tokenizer_config.json",
         help="Llama3.1 405b tokenizer path, defaults to 30F CI system path",
     )
 
@@ -148,7 +148,7 @@ def pytest_addoption(parser):
         "--llama3-405b-f16-model-path",
         type=Path,
         action="store",
-        default="/data/extra/models/llama3.1_405B/llama405b_fp16.irpa",
+        default="/data/llama-3.1/405b/llama405b_fp16.irpa",
         help="Llama3.1 405b model path, defaults to 30F CI system path",
     )