diff --git a/README.md b/README.md index 4e44889e66d7e3de1bb19f1c5c22cad141420095..83a634a9dc07780df3ee65469edd8e58e3ba9ba9 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,4 @@ -# Top-notch Transformers +# Lean, elegant, explorative Transformers This package implements Transformers using standard PyTorch code. Transformer models implement enhancements and encourage best-practices @@ -34,7 +34,7 @@ python -m pip install git+<repository-uri>@pytorch-2.0 ``` import torch -from tn_transformers import LPETransformer +from lee_transformers import LPETransformer batch_size = 2 @@ -64,7 +64,7 @@ assert outputs.shape == (batch_size, output_seq_len, num_outputs) ``` import torch -from tn_transformers import LPETransformer +from lee_transformers import LPETransformer batch_size = 2 @@ -90,7 +90,7 @@ assert outputs.shape == (batch_size, input_seq_len, num_outputs) ``` import torch -from tn_transformers import LPETransformer +from lee_transformers import LPETransformer batch_size = 2 @@ -117,7 +117,7 @@ assert outputs.shape == (batch_size, input_seq_len, num_outputs) ``` import torch -from tn_transformers import LPETransformer +from lee_transformers import LPETransformer batch_size = 2 diff --git a/docs/Makefile b/docs/Makefile index 80bde15bcfc86df500b1049671fa498ccce0c9df..e013d609e6756081b113ddaddadb05618d32dfa2 100644 --- a/docs/Makefile +++ b/docs/Makefile @@ -23,5 +23,5 @@ clean: # Catch-all target: route all unknown targets to Sphinx using the new # "make mode" option. $(O) is meant as a shortcut for $(SPHINXOPTS). %: Makefile - @$(SPHINXPREBUILD) -o "$(GENERATEDDIR)" "$(SOURCEDIR)"/../tn_transformers + @$(SPHINXPREBUILD) -o "$(GENERATEDDIR)" "$(SOURCEDIR)"/../lee_transformers @$(SPHINXBUILD) -M $@ "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O) diff --git a/docs/index.rst b/docs/index.rst index 7a31f8e1e729b7aaed3e9dec48166b7092bb9885..25811a446ee013d0b8826c44c2f716e224526c61 100644 --- a/docs/index.rst +++ b/docs/index.rst @@ -1,6 +1,6 @@ .. TN-Transformers documentation master file. -.. Assumes `sphinx-apidoc -o generated ../tn_transformers` to have been +.. Assumes `sphinx-apidoc -o generated ../lee_transformers` to have been ran from this directory. TN-Transformers diff --git a/tn_transformers/__init__.py b/lee_transformers/__init__.py similarity index 100% rename from tn_transformers/__init__.py rename to lee_transformers/__init__.py diff --git a/tn_transformers/attention_masks.py b/lee_transformers/attention_masks.py similarity index 100% rename from tn_transformers/attention_masks.py rename to lee_transformers/attention_masks.py diff --git a/tn_transformers/common.py b/lee_transformers/common.py similarity index 100% rename from tn_transformers/common.py rename to lee_transformers/common.py diff --git a/tn_transformers/initialization.py b/lee_transformers/initialization.py similarity index 100% rename from tn_transformers/initialization.py rename to lee_transformers/initialization.py diff --git a/tn_transformers/layers/__init__.py b/lee_transformers/layers/__init__.py similarity index 100% rename from tn_transformers/layers/__init__.py rename to lee_transformers/layers/__init__.py diff --git a/tn_transformers/layers/adaptive_computation_time.py b/lee_transformers/layers/adaptive_computation_time.py similarity index 100% rename from tn_transformers/layers/adaptive_computation_time.py rename to lee_transformers/layers/adaptive_computation_time.py diff --git a/tn_transformers/layers/glu_layers.py b/lee_transformers/layers/glu_layers.py similarity index 100% rename from tn_transformers/layers/glu_layers.py rename to lee_transformers/layers/glu_layers.py diff --git a/tn_transformers/layers/positional_encodings.py b/lee_transformers/layers/positional_encodings.py similarity index 100% rename from tn_transformers/layers/positional_encodings.py rename to lee_transformers/layers/positional_encodings.py diff --git a/tn_transformers/layers/resampling.py b/lee_transformers/layers/resampling.py similarity index 100% rename from tn_transformers/layers/resampling.py rename to lee_transformers/layers/resampling.py diff --git a/tn_transformers/layers/rms_norm.py b/lee_transformers/layers/rms_norm.py similarity index 100% rename from tn_transformers/layers/rms_norm.py rename to lee_transformers/layers/rms_norm.py diff --git a/tn_transformers/layers/rpe_mha.py b/lee_transformers/layers/rpe_mha.py similarity index 100% rename from tn_transformers/layers/rpe_mha.py rename to lee_transformers/layers/rpe_mha.py diff --git a/tn_transformers/layers/seq_pool.py b/lee_transformers/layers/seq_pool.py similarity index 100% rename from tn_transformers/layers/seq_pool.py rename to lee_transformers/layers/seq_pool.py diff --git a/tn_transformers/layers/utils.py b/lee_transformers/layers/utils.py similarity index 100% rename from tn_transformers/layers/utils.py rename to lee_transformers/layers/utils.py diff --git a/tn_transformers/models/__init__.py b/lee_transformers/models/__init__.py similarity index 100% rename from tn_transformers/models/__init__.py rename to lee_transformers/models/__init__.py diff --git a/tn_transformers/models/common.py b/lee_transformers/models/common.py similarity index 100% rename from tn_transformers/models/common.py rename to lee_transformers/models/common.py diff --git a/tn_transformers/models/lpe_transformer.py b/lee_transformers/models/lpe_transformer.py similarity index 100% rename from tn_transformers/models/lpe_transformer.py rename to lee_transformers/models/lpe_transformer.py diff --git a/tn_transformers/models/rpe_transformer.py b/lee_transformers/models/rpe_transformer.py similarity index 100% rename from tn_transformers/models/rpe_transformer.py rename to lee_transformers/models/rpe_transformer.py diff --git a/tn_transformers/models/transformer.py b/lee_transformers/models/transformer.py similarity index 100% rename from tn_transformers/models/transformer.py rename to lee_transformers/models/transformer.py diff --git a/tn_transformers/schedules.py b/lee_transformers/schedules.py similarity index 100% rename from tn_transformers/schedules.py rename to lee_transformers/schedules.py diff --git a/tn_transformers/typing.py b/lee_transformers/typing.py similarity index 100% rename from tn_transformers/typing.py rename to lee_transformers/typing.py diff --git a/setup.py b/setup.py index 2a39bb80d79a5d86b48d2bc37699394c6cedfde7..bbbb6c728c45332ca929d3b13fed81c1e3b15a8f 100644 --- a/setup.py +++ b/setup.py @@ -1,7 +1,7 @@ from setuptools import find_packages, setup setup( - name='tn_transformers', + name='lee_transformers', python_requires='>=3.8', version='0.1.6', packages=find_packages(), diff --git a/test/test_layers.py b/test/test_layers.py index cada1db62e5a7ca2f949dab78720ed531337a55a..cada158fbbaea84ae2897422e6443d5bde05a557 100644 --- a/test/test_layers.py +++ b/test/test_layers.py @@ -2,7 +2,7 @@ import unittest import torch as th -from tn_transformers.layers import ( +from lee_transformers.layers import ( AdaptiveComputationTime, DownscalingXPosEncoding, GBST, diff --git a/test/test_masks.py b/test/test_masks.py index a6f7166f28aaa3757e32a05abc6ff03bcf681c70..676497d33f9ca540e21658f3f3aa512eeee63018 100644 --- a/test/test_masks.py +++ b/test/test_masks.py @@ -2,7 +2,7 @@ import unittest import torch as th -from tn_transformers.attention_masks import ( +from lee_transformers.attention_masks import ( alibi, causal, full, diff --git a/test/test_models.py b/test/test_models.py index b837ffd7a35859c4e55801710e76029769767a67..7156ce1a6d0991c2940137f34bdff0ffcec6cdbc 100644 --- a/test/test_models.py +++ b/test/test_models.py @@ -2,8 +2,8 @@ import unittest import torch as th -from tn_transformers import LPETransformer, RPETransformer, Transformer -from tn_transformers.layers import PositionalEncoding +from lee_transformers import LPETransformer, RPETransformer, Transformer +from lee_transformers.layers import PositionalEncoding from .common import check_grad