[Core] Move ray_utils.py from engine to executor package (#4347)

This commit is contained in:
Nick Hill
2024-04-24 23:52:22 -07:00
committed by GitHub
parent 96e90fdeb3
commit 479d69fad0
7 changed files with 11 additions and 9 deletions

View File

@@ -10,7 +10,7 @@ from transformers import PreTrainedTokenizer
from vllm.config import ModelConfig
from vllm.engine.arg_utils import AsyncEngineArgs
from vllm.engine.llm_engine import LLMEngine
from vllm.engine.ray_utils import initialize_ray_cluster, ray
from vllm.executor.ray_utils import initialize_ray_cluster, ray
from vllm.logger import init_logger
from vllm.lora.request import LoRARequest
from vllm.outputs import RequestOutput