#9604: Update Mixtral demo inputs and expected outputs for token matc…

…hing. Relaxed Mixtral perf estimates due to CI variability. Minor fix to demo CI
tenstorrent · Jul 3, 2024 · ce1e177 · ce1e177
1 parent 3521f5e
commit ce1e177
Show file tree

Hide file tree

Showing 4 changed files with 76 additions and 77 deletions.
diff --git a/models/demos/t3000/mixtral8x7b/demo/demo.py b/models/demos/t3000/mixtral8x7b/demo/demo.py
@@ -266,7 +266,7 @@ def run_mixtral_demo(user_input, batch_size, device_mesh, instruct_mode):
 
 
 # Avoid running this test when in CI
-@pytest.mark.skipif(os.getenv("CI") == "True", reason="Non-CI tests")
+@pytest.mark.skipif(os.getenv("CI") == "true", reason="Non-CI tests")
 @pytest.mark.parametrize(
     "input_prompts, instruct_weights",
     [
@@ -282,7 +282,7 @@ def test_mixtral8x7b_demo(t3k_device_mesh, use_program_cache, input_prompts, ins
 
 
 # CI only runs general-weights demo
-@pytest.mark.skipif(os.getenv("CI") == "False", reason="CI-only test")
+@pytest.mark.skipif(not os.getenv("CI") == "true", reason="CI-only test")
 @pytest.mark.parametrize(
     "input_prompts, instruct_weights",
     [