fix pylint

alibaba · Jan 10, 2025 · b651012 · b651012
1 parent edaa0c6
commit b651012
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/chatlearn/models/vllm_module_v2.py b/chatlearn/models/vllm_module_v2.py
@@ -88,10 +88,6 @@ def setup(self):
         tokenizer.tokenizer = tokenizer
         self.tokenizer = tokenizer
 
-    def model_setup(self):
-        """Set up model and enable EMS(Efficient Memory Sharing)"""
-        super().model_setup()
-
     def setup_vllm(self, workers):
         if self.llm is not None: # for evaluator
             return
@@ -345,6 +341,7 @@ def pipeline_parallel_rank(self):
     def model_setup_for_workers(self):
         self.llm.llm_engine.model_executor._run_workers("model_setup")
 
+    # pylint: disable=unused-argument
     def offload_for_workers(self,
                             to_onload_weights=None,
                             to_build_grad_buffers=None,