diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-09-05 16:23:03 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-09-05 16:23:03 -0700 |
| commit | eab4d8232d558f2e6bd7f7cc3d00e2e6e94f4e80 (patch) | |
| tree | 1a70991e0d9d5632a20388bf87e4c56dfb06d64e /bitsandbytes/optim/__init__.py | |
| parent | 5d9bc7a301618d6642a70951ac46637d4889c82b (diff) | |
| parent | 8e7053a988f7bf6c0b2605a3d0ec461f5d9b05e4 (diff) | |
Merge branch 'bugfix_cpu_and_cc' into main
Diffstat (limited to 'bitsandbytes/optim/__init__.py')
| -rw-r--r-- | bitsandbytes/optim/__init__.py | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/bitsandbytes/optim/__init__.py b/bitsandbytes/optim/__init__.py index a76d717..d18f1d1 100644 --- a/bitsandbytes/optim/__init__.py +++ b/bitsandbytes/optim/__init__.py @@ -5,13 +5,12 @@ from bitsandbytes.cextension import COMPILED_WITH_CUDA -if COMPILED_WITH_CUDA: - from .adam import Adam, Adam8bit, Adam32bit - from .adamw import AdamW, AdamW8bit, AdamW32bit - from .sgd import SGD, SGD8bit, SGD32bit - from .lars import LARS, LARS8bit, LARS32bit, PytorchLARS - from .lamb import LAMB, LAMB8bit, LAMB32bit - from .rmsprop import RMSprop, RMSprop8bit, RMSprop32bit - from .adagrad import Adagrad, Adagrad8bit, Adagrad32bit +from .adam import Adam, Adam8bit, Adam32bit +from .adamw import AdamW, AdamW8bit, AdamW32bit +from .sgd import SGD, SGD8bit, SGD32bit +from .lars import LARS, LARS8bit, LARS32bit, PytorchLARS +from .lamb import LAMB, LAMB8bit, LAMB32bit +from .rmsprop import RMSprop, RMSprop8bit, RMSprop32bit +from .adagrad import Adagrad, Adagrad8bit, Adagrad32bit from .optimizer import GlobalOptimManager |