fix: return loaded parameters in Aria load_weights (#57476)
Signed-off-by: Debanshu Das <debanshu@google.com>
debanshd committed
Sep 28, 2026 at 21:05 UTC
13c8589b8650f291223cb20a41abe9957b1c5e96
1 file changed
+5
-2
vllm/model_executor/models/aria.py
+5
-2
@@ -526,6 +526,9 @@ class AriaForConditionalGeneration(nn.Module, SupportsMultiModal):
526
logits = self.logits_processor(self.lm_head, hidden_states)
527
return logits
528
529
- def load_weights(self, weights: Iterable[tuple[str, torch.Tensor]]):
529
+ def load_weights(
530
+ self,
531
+ weights: Iterable[tuple[str, torch.Tensor]],
532
+ ) -> set[str]:
533
loader = AutoWeightsLoader(self)
531
- loader.load_weights(weights, mapper=self.hf_to_vllm_mapper)
534
+ return loader.load_weights(weights, mapper=self.hf_to_vllm_mapper)