[Bugfix] Fix typo in FusedMoE LoRA reshape comment (#31992)

Signed-off-by: Xin Yang <xyangx@amazon.com>
This commit is contained in:
Xin Yang
2026-01-08 18:46:05 -08:00
committed by GitHub
parent 8ff4a99566
commit 8413868dab

View File

@@ -632,7 +632,7 @@ class LoRAModelManager:
num_experts, -1, down_proj_lora.lora_a.shape[-1]
)
# (output_size,num_experts,rank)
# (output_size,rank,num_experts)
gate_up_proj_lora.lora_b = gate_up_proj_lora.lora_b.reshape(
gate_up_proj_lora.lora_b.shape[0], -1, num_experts
)