forked from mrq/bitsandbytes-rocm
Version bump + bnb.utils import fix.
This commit is contained in:
parent
f0ae860c86
commit
2e630b55f5
|
@ -12,7 +12,7 @@ from .autograd._functions import (
|
|||
)
|
||||
from .cextension import COMPILED_WITH_CUDA
|
||||
from .nn import modules
|
||||
from . import cuda_setup
|
||||
from . import cuda_setup, utils
|
||||
|
||||
if COMPILED_WITH_CUDA:
|
||||
from .optim import adam
|
||||
|
|
Loading…
Reference in New Issue
Block a user