[Qwen2.5] Support bnb quant for Qwen2.5 (vllm-project#9467)

Signed-off-by: Maxime Fournioux <[email protected]>
mfournioux · Nov 20, 2024 · 297b83d · 297b83d
1 parent 0a5cf84
commit 297b83d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/vllm/model_executor/models/qwen2.py b/vllm/model_executor/models/qwen2.py
@@ -364,6 +364,14 @@ class Qwen2ForCausalLM(nn.Module, SupportsLoRA, SupportsPP):
     ]
     embedding_modules = {}
     embedding_padding_modules = []
+    bitsandbytes_stacked_params_mapping = {
+        # shard_name, weight_name, index
+        "q_proj": ("qkv_proj", 0),
+        "k_proj": ("qkv_proj", 1),
+        "v_proj": ("qkv_proj", 2),
+        "gate_proj": ("gate_up_proj", 0),
+        "up_proj": ("gate_up_proj", 1),
+    }
 
     def __init__(
         self,