From b65101271d600720f5b0a34f7b8badab57f22abd Mon Sep 17 00:00:00 2001 From: "wanglin.zj" Date: Fri, 10 Jan 2025 16:32:35 +0800 Subject: [PATCH] fix pylint --- chatlearn/models/vllm_module_v2.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/chatlearn/models/vllm_module_v2.py b/chatlearn/models/vllm_module_v2.py index dad60f2a..8e128c3b 100644 --- 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,