Only fix signature

Author: BlackHC

src/lightning/pytorch/core/module.py

@@ -675,6 +675,17 @@ def all_gather(
         all_gather = self.trainer.strategy.all_gather
         data = convert_to_tensors(data, device=self.device)
         return apply_to_collection(data, Tensor, all_gather, group=group, sync_grads=sync_grads)
+        
+    @override
+    def forward(self, *inputs) -> Any:
+        r"""Same as :meth:`torch.nn.Module.forward`.
+        Args:
+            *args: Whatever you decide to pass into the forward method.
+            **kwargs: Keyword arguments are also possible.
+        Return:
+            Your model's output
+        """
+        return super().forward(*inputs)
 
     def training_step(self, *args: Any, **kwargs: Any) -> STEP_OUTPUT:
         r"""Here you compute and return the training loss and some additional metrics for e.g. the progress bar or