update ray.init()

tests/utils.py

@@ -340,7 +340,7 @@ def multi_process_parallel(
     # as compared to multiprocessing.
     # NOTE: We need to set working_dir for distributed tests,
     # 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()
     refs = []