From 7e2fbc1df701ccad6ba9d6fab44dff53593f1c36 Mon Sep 17 00:00:00 2001
From: Alexandru Gherghescu <gherghescu_alex1@yahoo.ro>
Date: Tue, 4 Jun 2024 00:25:39 +0300
Subject: [PATCH] Remove useless models init file
---
optimus/models/__init__.py | 2 --
optimus/models/pytorch_transformer.py | 2 +-
2 files changed, 1 insertion(+), 3 deletions(-)
delete mode 100644 optimus/models/__init__.py
diff --git a/optimus/models/__init__.py b/optimus/models/__init__.py
deleted file mode 100644
index d38f836..0000000
--- 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 ccc81b4..62cadc0 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
--
GitLab