forked from mrq/bitsandbytes-rocm
Merge branch 'main' of github.com:TimDettmers/bitsandbytes into main
This commit is contained in:
commit
8ccc0b0ee1
|
@ -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