Init data ckpt for VLLMModuleV2. (#186)

alibaba · Dec 24, 2024 · 36c2794 · 36c2794
1 parent 1081492
commit 36c2794
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chatlearn/runtime/dist_actor.py b/chatlearn/runtime/dist_actor.py
@@ -255,7 +255,7 @@ def add_remote_func(self):
             setattr(self, func_name, dist_call)
 
     def model_setup(self):
-        return [self.vllm_engine.setup.remote()]
+        return [self.vllm_engine.model_setup.remote()]
 
     @property
     def master(self):