Skip to content

Commit

Permalink
#13368: Remove unused parameter from model demo
Browse files Browse the repository at this point in the history
  • Loading branch information
mtairum committed Oct 16, 2024
1 parent bdf54f3 commit 53d150a
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions models/demos/llama3/demo/demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -506,14 +506,12 @@ def run_llama_demo_n300(user_input, batch_size, mesh_device, instruct_mode, is_c
("models/demos/llama3/demo/input_data_prefill_128.json", False, 3),
("models/demos/llama3/demo/input_data_questions_prefill_128.json", True, 1),
("models/demos/llama3/demo/input_data_questions_prefill_128.json", True, 3),
("models/demos/llama3/demo/input_data_long.json", True, 1),
],
ids=[
"general_weights-1_batch",
"general_weights-3_batch",
"instruct_weights-1_batch",
"instruct_weights-3_batch",
"instruct_weights-long",
],
)
@pytest.mark.parametrize("device_params", [{"trace_region_size": 5560320, "num_command_queues": 2}], indirect=True)
Expand Down

0 comments on commit 53d150a

Please sign in to comment.