diff --git a/optimus/models/__init__.py b/optimus/models/__init__.py
deleted file mode 100644
index d38f83611e34610eacba0f3de7b07344e9e7ed73..0000000000000000000000000000000000000000
--- a/optimus/models/__init__.py
+++ /dev/null
@@ -1,2 +0,0 @@
-from .optimus import OptimusTransformer
-from .pytorch_transformer import PyTorchTransformer
diff --git a/optimus/models/pytorch_transformer.py b/optimus/models/pytorch_transformer.py
index ccc81b4bde96e70b585a145a0ad560e8543cd954..62cadc01c34eef58eac74b5afaeca4fc8ca44e68 100644
--- a/optimus/models/pytorch_transformer.py
+++ b/optimus/models/pytorch_transformer.py
@@ -134,4 +134,4 @@ class PyTorchTransformer(nn.Module):
             x = self.output_norm(x)
             output = self.output(output)
 
-            return output
\ No newline at end of file
+            return output