Update path for local irpa file

nod-ai · Jan 24, 2025 · 488bc64 · 488bc64
1 parent d06c9b9
commit 488bc64
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app_tests/integration_tests/llm/shortfin/conftest.py b/app_tests/integration_tests/llm/shortfin/conftest.py
@@ -27,8 +27,8 @@
     ),
     "llama3.1_8b": ModelConfig(
         source=ModelSource.LOCAL,
-        local_path=Path("/data/llama3.1/8b/llama8b_f16.irpa"),
-        model_file="llama8b_f16.irpa",
+        local_path=Path("/data/llama3.1/weights/8b/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),
         device_settings=device_settings.CPU,