소스 검색

fix ray gpu executor

AlpinDale 7 달 전
부모
커밋
9f3d6205ce
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      aphrodite/executor/ray_gpu_executor.py

+ 1 - 1
aphrodite/executor/ray_gpu_executor.py

@@ -101,7 +101,7 @@ class RayGPUExecutor(DistributedGPUExecutor):
                 **ray_remote_kwargs,
             )(RayWorkerWrapper).remote(
                 worker_module_name=worker_module_name,
-                worker_class_name="worker_class_name",
+                worker_class_name=worker_class_name,
                 trust_remote_code=self.model_config.trust_remote_code,
             )