Skip to content

GitLab

Unverified Commit 304d5ba1 authored by Nicolò Lucchesi's avatar Nicolò Lucchesi Committed by GitHub
Browse files

[Bugfix][CI] Fix `tests/distributed/test_torchrun_example_moe.py` (#40349) 


Signed-off-by: default avatarNickLucche <nlucches@redhat.com>
parent 81d954f4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tests/distributed/test_torchrun_example_moe.py
View file @ 304d5ba1
...@@ -38,6 +38,8 @@ llm = LLM( ...@@ -38,6 +38,8 @@ llm = LLM(
distributed_executor_backend="external_launcher", distributed_executor_backend="external_launcher",
gpu_memory_utilization=random.uniform(0.7, 0.9), gpu_memory_utilization=random.uniform(0.7, 0.9),
seed=0, seed=0,
max_model_len=1024,
max_num_seqs=16,
) )
outputs = llm.generate(prompts, sampling_params) outputs = llm.generate(prompts, sampling_params)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment