From 56090fb1dfc7ccdbe7ab106134958ba7ec9f0114 Mon Sep 17 00:00:00 2001
From: xffxff <1247714429@qq.com>
Date: Fri, 22 Nov 2024 03:58:06 +0000
Subject: [PATCH] add some comments

Signed-off-by: xffxff <1247714429@qq.com>
---
 vllm/model_executor/models/aria.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/vllm/model_executor/models/aria.py b/vllm/model_executor/models/aria.py
index dd2ca3e87c3b0..dcad38ef454ea 100644
--- a/vllm/model_executor/models/aria.py
+++ b/vllm/model_executor/models/aria.py
@@ -471,6 +471,8 @@ def _make_expert_params_mapping(
             ]
         ]
 
+    # Adapted from LlamaModel.load_weights with the modification of adding the
+    # expert_params_mapping
     def load_weights(self, weights: Iterable[Tuple[str,
                                                    torch.Tensor]]) -> Set[str]:
         stacked_params_mapping = [