Fix path to local Llama file

nod-ai · Jan 24, 2025 · 7024793 · 7024793
1 parent 5777acc
commit 7024793
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/app_tests/integration_tests/llm/shortfin/conftest.py b/app_tests/integration_tests/llm/shortfin/conftest.py
@@ -27,7 +27,9 @@
     ),
     "llama3.1_8b": ModelConfig(
         source=ModelSource.LOCAL,
-        local_path=Path("/data/llama3.1/weights/8b/llama3.1_8b_fp16_instruct.irpa"),
+        local_path=Path(
+            "/data/llama3.1/weights/8b/fp16/llama3.1_8b_fp16_instruct.irpa"
+        ),
         model_file="llama3.1_8b_fp16_instruct.irpa",
         tokenizer_id="NousResearch/Meta-Llama-3.1-8B",
         batch_sizes=(1, 4),