Skip to content

Commit

Permalink
#13368: Add reset seeds to llama3 demo
Browse files Browse the repository at this point in the history
  • Loading branch information
mtairum committed Oct 30, 2024
1 parent 3662095 commit 85903e2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion models/demos/llama3/demo/demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)")
Expand Down

0 comments on commit 85903e2

Please sign in to comment.