Fix mypy error

stubs/torch/__init__.pyi

@@ -25,6 +25,11 @@ from .autograd import no_grad as no_grad, enable_grad as enable_grad, \
 from . import cuda as cuda
 from . import optim as optim
 
+#MODIFIED BY TORCHGPIPE
+from . import version
+from . import backends
+#END
+
 class dtype: ...
 
 class layout: ...

stubs/torch/backends/__init__.pyi

@@ -0,0 +1,3 @@
+#MODIFIED BY TORCHGPIPE
+from . import cudnn
+#END

stubs/torch/backends/cudnn.pyi

@@ -0,0 +1,3 @@
+#MODIFIED BY TORCHGPIPE
+def version() -> int: ...
+#END

stubs/torch/nn/__init__.pyi

@@ -52,9 +52,6 @@ class Module:
 
     def add_module(self, name: str, module: Module) -> None: ...
 
-    def buffers(self) -> Iterator[Tensor]: ...
-    def parameters(self) -> Iterator[Parameter]: ...
-
     def state_dict(self, destination: Optional[str] = ..., prefix: str = ..., keep_vars: bool = ...) -> Dict[str, Tensor]: ...
     def load_state_dict(self, state_dict: Dict[str, Tensor], strict: bool = ...) -> Tuple[List[str], List[str]]: ...
 

stubs/torch/version.pyi

@@ -0,0 +1,5 @@
+#MODIFIED BY TORCHGPIPE
+debug: bool = ...
+cuda: str = ...
+git_version: str = ...
+#END