diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2021-11-29 09:32:13 -0800 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2021-11-29 09:32:13 -0800 |
| commit | 20e1677dfdc4495038fd780807c8cbc253adf921 (patch) | |
| tree | 42011169e55eab3f4226ff171d84edac84ec6f8f /bitsandbytes/nn/__init__.py | |
| parent | 3cff6795fb70dd99b4802593f3c70d291e0cd1dc (diff) | |
Added module override, bnb.nn.Embedding #13 #15 #19
Diffstat (limited to 'bitsandbytes/nn/__init__.py')
| -rw-r--r-- | bitsandbytes/nn/__init__.py | 2 |
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 |