bitsandbytes-rocm/bitsandbytes
Tim Dettmers 9d353ca786
Merge pull request #87 from lostmsu/main
Add `device` and `dtype` parameters to `StableEmbedding`
2023-01-02 13:22:45 +01:00
..
autograd Resolve dangerous default value [] as argument 2022-10-27 13:25:51 +02:00
cuda_setup Default to CPU library on CUDA error+small refactor. 2023-01-02 03:47:09 -08:00
nn Merge pull request #87 from lostmsu/main 2023-01-02 13:22:45 +01:00
optim Fix several typos in logging and comments 2022-11-01 09:53:47 +01:00
__init__.py Sort imports 2022-10-27 13:15:21 +02:00
__main__.py Sort imports 2022-10-27 13:15:21 +02:00
cextension.py CUDASetup only executed once + fixed circular import. 2023-01-02 03:31:43 -08:00
functional.py Merge branch 'main' into cleanup 2023-01-02 11:19:16 +01:00
utils.py Replace print_stderr with warnings.warn 2022-08-24 18:45:17 +03:00