Fix artifacts_dir path

Signed-off-by: aviator19941 <[email protected]>
nod-ai · Oct 31, 2024 · e9ad709 · e9ad709
1 parent d683163
commit e9ad709
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sharktank/tests/models/llama/benchmark_amdgpu_test.py b/sharktank/tests/models/llama/benchmark_amdgpu_test.py
@@ -56,7 +56,7 @@ class BenchmarkLlama3_1_8B(BaseBenchmarkTest):
     def setUp(self):
         super().setUp()
         # TODO: add numpy files to Azure and download from it
-        self.artifacts_dir = Path("/data/extra/models/llama3.1_8B")
+        self.artifacts_dir = Path("/data/llama-3.1/8b")
         self.gguf_path = self.artifacts_dir / "llama8b_f16.gguf"
         self.irpa_path = self.artifacts_dir / "llama8b_f16.irpa"
         self.irpa_path_fp8 = self.artifacts_dir / "llama8b_fp8.irpa"