Skip to content

GitLab

Commit 63f5f03a authored by zhuwenwen's avatar zhuwenwen
Browse files

update ray.init()

parent da99b8e3
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/utils.py
View file @ 63f5f03a
...@@ -340,7 +340,7 @@ def multi_process_parallel( ...@@ -340,7 +340,7 @@ def multi_process_parallel(
# as compared to multiprocessing. # as compared to multiprocessing.
# NOTE: We need to set working_dir for distributed tests, # NOTE: We need to set working_dir for distributed tests,
# otherwise we may get import errors on ray workers # otherwise we may get import errors on ray workers
ray.init(runtime_env={"working_dir": VLLM_PATH}) ray.init(num_gpus=tp_size, runtime_env={"working_dir": VLLM_PATH})
distributed_init_port = get_open_port() distributed_init_port = get_open_port()
refs = [] refs = []
......
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