From 037bcd942cac24a30dfe61ea3423caf62c92a153 Mon Sep 17 00:00:00 2001 From: Mrm <86636997+noc-turne@users.noreply.github.com> Date: Mon, 31 Mar 2025 21:56:42 +0800 Subject: [PATCH] [Bugfix] Fix missing return value in load_weights method of adapters.py (#15542) Signed-off-by: noc-turne <2270929247@qq.com> --- vllm/model_executor/models/adapters.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/vllm/model_executor/models/adapters.py b/vllm/model_executor/models/adapters.py index 23d72d8e60f6..6ab03c40ab4a 100644 --- a/vllm/model_executor/models/adapters.py +++ b/vllm/model_executor/models/adapters.py @@ -99,16 +99,17 @@ def _create_pooling_model_cls( mapper = WeightsMapper(orig_to_new_prefix={"model.": ""}) weights = mapper.apply(weights) - self.model.load_weights(weights) - return + loaded_params = self.model.load_weights(weights) + loaded_params = {f"model.{name}" for name in loaded_params} + return loaded_params # For most other models if hasattr(orig_cls, "load_weights"): - orig_cls.load_weights(self, weights) # type: ignore + return orig_cls.load_weights(self, weights) # type: ignore # Fallback else: loader = AutoWeightsLoader(self) - loader.load_weights(weights) + return loader.load_weights(weights) return ModelForPooling # type: ignore