[Bugfix] Fix FusedMoE LoRA w2_output_size (#31949)

Signed-off-by: Xin Yang <xyangx@amazon.com>

[Bugfix] Fix FusedMoE LoRA w2_output_size (#31949)
Signed-off-by: Xin Yang <xyangx@amazon.com>
707b240d · Xin Yang · GitHub · 29ce4822 · 707b240d
Unverified Commit 707b240d authored Jan 08, 2026 by Xin Yang Committed by GitHub Jan 09, 2026
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

vllm/lora/layers/fused_moe.py vllm/lora/layers/fused_moe.py +1 -1

No files found.
--- a/vllm/lora/layers/fused_moe.py
+++ b/vllm/lora/layers/fused_moe.py
@@ -720,7 +720,7 @@ class FusedMoE3DWithLoRA(FusedMoEWithLoRA):
        """
        Full size
        """
-        return self.w2_lora_a_stacked[0].shape[-2]
+        return self.base_layer.hidden_size

    @classmethod
    def can_replace_layer(