summaryrefslogtreecommitdiff
path: root/bitsandbytes/nn/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'bitsandbytes/nn/__init__.py')
-rw-r--r--bitsandbytes/nn/__init__.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/bitsandbytes/nn/__init__.py b/bitsandbytes/nn/__init__.py
index 177540f..27ad6ca 100644
--- a/bitsandbytes/nn/__init__.py
+++ b/bitsandbytes/nn/__init__.py
@@ -2,4 +2,4 @@
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
-from .modules import StableEmbedding
+from .modules import StableEmbedding, Embedding