#0: Update lt to use the new demo

tenstorrent · Jan 17, 2025 · decc3fa · decc3fa
1 parent 1075e20
commit decc3fa
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/models/demos/llama3/lt b/models/demos/llama3/lt
@@ -827,10 +827,14 @@ def run_entry_command(entry, screen_lock, output_entries, screen_needs_update):
     command_shortcuts = {
         "accuracy": "pytest models/demos/llama3/tests/test_llama_accuracy.py -k 'attention-performance and file'",
         "accuracy-acc": "pytest models/demos/llama3/tests/test_llama_accuracy.py -k 'attention-acc and file'",
-        "demo": "pytest models/demos/llama3/demo/demo.py -k performance-batch-1",
-        "demo-acc": "pytest models/demos/llama3/demo/demo.py -k accuracy-batch-1",
-        "demo-32": "pytest models/demos/llama3/demo/demo.py -k performance-batch-32",
-        "demo-long": "pytest models/demos/llama3/demo/demo.py -k performance-long",
+        "demo": "pytest models/demos/llama3/demo/simple_text_demo.py -k performance-batch-1",
+        "demo-acc": "pytest models/demos/llama3/demo/simple_text_demo.py -k accuracy-batch-1",
+        "demo-32": "pytest models/demos/llama3/demo/simple_text_demo.py -k performance-batch-32",
+        "demo-long": "pytest models/demos/llama3/demo/simple_text_demo.py -k performance-long",
+        "leg-demo": "pytest models/demos/llama3/demo/demo.py -k performance-batch-1",
+        "leg-demo-acc": "pytest models/demos/llama3/demo/demo.py -k accuracy-batch-1",
+        "leg-demo-32": "pytest models/demos/llama3/demo/demo.py -k performance-batch-32",
+        "leg-demo-long": "pytest models/demos/llama3/demo/demo.py -k performance-long",
         "attention": "pytest models/demos/llama3/tests/test_llama_attention.py",
         "attention-prefill": "pytest models/demos/llama3/tests/test_llama_attention_prefill.py",
         "mlp": "pytest models/demos/llama3/tests/test_llama_mlp.py",