diff --git a/pyproject.toml b/pyproject.toml index 0134a30..14b74db 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -14,7 +14,7 @@ train = [ "transformers>=4.40", "trl>=0.8", "torch>=2.0", - "bitsandbytes>=0.43", + "bitsandbytes>=0.49.2", ] dev = [ "pytest>=8.0",