From 13c8589b8650f291223cb20a41abe9957b1c5e96 Mon Sep 17 00:00:00 2001 From: debanshd Date: Mon, 28 Sep 2026 21:05:07 -0700 Subject: [PATCH] fix: return loaded parameters in Aria load_weights (#57476) Signed-off-by: Debanshu Das --- vllm/model_executor/models/aria.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/vllm/model_executor/models/aria.py b/vllm/model_executor/models/aria.py index da91c9d9bf..f063f12fdb 100644 --- a/vllm/model_executor/models/aria.py +++ b/vllm/model_executor/models/aria.py @@ -526,6 +526,9 @@ class AriaForConditionalGeneration(nn.Module, SupportsMultiModal): logits = self.logits_processor(self.lm_head, hidden_states) return logits - def load_weights(self, weights: Iterable[tuple[str, torch.Tensor]]): + def load_weights( + self, + weights: Iterable[tuple[str, torch.Tensor]], + ) -> set[str]: loader = AutoWeightsLoader(self) - loader.load_weights(weights, mapper=self.hf_to_vllm_mapper) + return loader.load_weights(weights, mapper=self.hf_to_vllm_mapper)