From 85903e2f2f1fa1cf36ebf4a8e181f1f47f66b064 Mon Sep 17 00:00:00 2001 From: mtairum Date: Wed, 30 Oct 2024 17:47:32 +0000 Subject: [PATCH] #13368: Add reset seeds to llama3 demo --- models/demos/llama3/demo/demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/demos/llama3/demo/demo.py b/models/demos/llama3/demo/demo.py index 6382222f815..d67520b2e85 100644 --- a/models/demos/llama3/demo/demo.py +++ b/models/demos/llama3/demo/demo.py @@ -727,7 +727,7 @@ def run_llama3_demo( indirect=True, ) def test_llama_demo( - mesh_device, use_program_cache, input_prompts, instruct_weights, is_ci_env, num_batches, single_layer + mesh_device, use_program_cache, input_prompts, instruct_weights, is_ci_env, num_batches, single_layer, reset_seeds ): if is_ci_env and (instruct_weights == False or "long" in input_prompts or single_layer == True): pytest.skip("CI demo test only runs instruct weights to reduce CI pipeline load (both are supported)")