diff --git a/.github/workflows/ci_eval.yaml b/.github/workflows/ci_eval.yaml index 27b6c94e6..5ac553052 100644 --- 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 index 026424693..f9ba99b2f 100644 --- 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", )