|
|
@@ -32,7 +32,7 @@ packages = find:
|
|
|
python_requires = >=3.8
|
|
|
install_requires =
|
|
|
torch>=1.12
|
|
|
- bitsandbytes==0.40.1.post1
|
|
|
+ bitsandbytes @ git+https://github.com/Titaniumtown/bitsandbytes-rocm@patch-2
|
|
|
accelerate>=0.20.3,<0.21.0
|
|
|
huggingface-hub>=0.11.1,<1.0.0
|
|
|
tokenizers>=0.13.3
|
Aleksandr Borzunov