fix hardcoded path in debug_inference_engine

4d962ffcaaeeb74a84027e7db13337fd313689f5 · 2024-07-19 23:44:42 -0700 · Alex Cheema

Files touched

M exo/inference/debug_inference_engine.py

Diff

commit 4d962ffcaaeeb74a84027e7db13337fd313689f5
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Fri Jul 19 23:44:42 2024 -0700

    fix hardcoded path in debug_inference_engine
---
 exo/inference/debug_inference_engine.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/exo/inference/debug_inference_engine.py b/exo/inference/debug_inference_engine.py
index 036a5879..b90044aa 100644
--- a/exo/inference/debug_inference_engine.py
+++ b/exo/inference/debug_inference_engine.py
@@ -39,5 +39,5 @@ async def test_inference_engine(inference_engine_1: InferenceEngine, inference_e
 asyncio.run(test_inference_engine(
     TinygradDynamicShardInferenceEngine(),
     TinygradDynamicShardInferenceEngine(),
-    "/Users/alex/Library/Caches/tinygrad/downloads/llama3-8b-sfr",
+    "llama3-8b-sfr",
 ))

← 30ab126c fix test_inference_engine · back to Exo · integration test with discovery 8dd17fe0 →