clean code

Signed-off-by: Yu Gong <yu3.gong@gmail.com>

Author: yugong333

vllm/lora/layers/fused_moe.py

@@ -136,7 +136,7 @@ def wrapper(*args, **kwargs):
                 M = min(num_tokens, CHUNK_SIZE)
 
                 shrink_config, expand_config = self._get_lora_moe_configs(
-                    op_prefix="gate_up",
+                    op_prefix="w13",
                     lora_a_stacked=self.w1_lora_a_stacked,
                     lora_b_stacked=self.w1_lora_b_stacked,
                     num_slices=2,
@@ -214,7 +214,7 @@ def wrapper(*args, **kwargs):
                 M = min(num_tokens, CHUNK_SIZE)
 
                 shrink_config, expand_config = self._get_lora_moe_configs(
-                    op_prefix="down",
+                    op_prefix="w2",
                     lora_a_stacked=self.w2_lora_a_stacked,
                     lora_b_stacked=self.w2_lora_b_stacked,
                     num_slices=1,