format code

Author: yao-fengchen

lmdeploy/pytorch/kernels/ascend/apply_rotary_pos_emb.py

@@ -26,7 +26,7 @@ def apply_rotary_pos_emb(
             cos = cos[position_ids_1d].view(1, bs, 1, -1)
             sin = sin[position_ids_1d].view(1, bs, 1, -1)
         else:
-            raise RuntimeError("Cannot handle cos/sin shape dims!")
+            raise RuntimeError('Cannot handle cos/sin shape dims!')
 
         if context:
             setattr(context, 'cos', cos)

lmdeploy/pytorch/models/qwen2_moe.py

@@ -106,10 +106,9 @@ def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
             routing_weights /= routing_weights.sum(dim=-1, keepdim=True)
         routing_weights = routing_weights.to(hidden_states.dtype)
 
-        out_states = torch.zeros(
-            (batch_size * sequence_length, hidden_dim),
-            dtype=hidden_states.dtype,
-            device=hidden_states.device)
+        out_states = torch.zeros((batch_size * sequence_length, hidden_dim),
+                                 dtype=hidden_states.dtype,
+                                 device=hidden_states.device)
 
         expert_mask = torch.nn.functional.one_hot(
             selected_experts, num_classes=self.num_experts).permute(2, 1, 0)