annotate input_

Signed-off-by: lucast2021 <[email protected]>
vllm-project · Jan 5, 2025 · 58fbdfb · 58fbdfb
1 parent 0c6272d
commit 58fbdfb
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/vllm/lora/layers.py b/vllm/lora/layers.py
@@ -406,8 +406,8 @@ def __init__(self, base_layer: ReplicatedLinear) -> None:
         self.n_slices = 1
 
     def forward(
-            self,
-            input_) -> Tuple[Optional[torch.Tensor], Optional[torch.Tensor]]:
+        self, input_: torch.Tensor
+    ) -> Tuple[Optional[torch.Tensor], Optional[torch.Tensor]]:
         """Forward of ReplicatedLinearWithLoRA
 
         Args:
@@ -499,8 +499,8 @@ def slice_bias(self, bias: torch.Tensor) -> torch.Tensor:
         return bias
 
     def forward(
-            self,
-            input_) -> Tuple[Optional[torch.Tensor], Optional[torch.Tensor]]:
+        self, input_: torch.Tensor
+    ) -> Tuple[Optional[torch.Tensor], Optional[torch.Tensor]]:
         """Forward of ColumnParallelLinear
 
         Args:
@@ -838,8 +838,8 @@ def slice_bias(self, bias: torch.Tensor) -> torch.Tensor:
         return bias
 
     def forward(
-            self,
-            input_) -> Tuple[Optional[torch.Tensor], Optional[torch.Tensor]]:
+        self, input_: torch.Tensor
+    ) -> Tuple[Optional[torch.Tensor], Optional[torch.Tensor]]:
         """Forward of RowParallelLinear
 
         Args: