adding support for int8 lora training

2026-09-10 12:20:21 +08:00 · 2023-01-25 04:19:23 +05:30
parent ff8a5b9a69
commit d75746be70
2 changed files with 41 additions and 1 deletions
@@ -43,6 +43,7 @@ setup(
        "torch>=1.13.0",
        "transformers",
        "accelerate",
+        "bitsandbytes",
    ],
    extras_require=extras,
    classifiers=[