fix typing

Author: wangzaijun

lightllm/models/qwen3_moe/layer_infer/transformer_layer_infer.py

@@ -124,7 +124,7 @@ def _tpsp_get_qkv(
         if infer_state.need_dp_prefill_balance:
             q = infer_state._all_to_all_unbalance_get(data=q)
             cache_kv = infer_state._all_to_all_unbalance_get(data=cache_kv)
-        
+
         return q, cache_kv
 
     def _moe_ffn(

lightllm/models/stablelm/layer_infer/transformer_layer_infer.py

@@ -54,7 +54,6 @@ def _tpsp_get_o(self, input, infer_state, layer_weight) -> Tuple[torch.Tensor, t
         # TODO
         raise Exception("not impl")
 
-
     def _att_norm(
         self, input, infer_state: LlamaInferStateInfo, layer_weight: StablelmTransformerLayerWeight
     ) -> torch.Tensor: