Updates after testing

2025-04-01 11:49:21 +00:00
parent 6f3d414d17
commit 130afbdd92
4 changed files with 28 additions and 11 deletions
--- a/tools/train/setup.py
+++ b/tools/train/setup.py
@@ -10,7 +10,7 @@ setup(

    install_requires=[
        'accelerate>=0.25.0',
-        'bitsandbytes>=0.41.1',
+        'bitsandbytes>=0.45.0',
        'black>=22.0.0',
        'datasets>=2.14.6',
        'einops>=0.7.0',