Update requirements to have bitsandbytes min and dolomite min (#153)

* Update requirements-cuda.txt to have bitsandbytes min Signed-off-by: Mustafa Eyceoz <[email protected]> * Update requirements.txt for dolomite min Signed-off-by: Mustafa Eyceoz <[email protected]> --------- Signed-off-by: Mustafa Eyceoz <[email protected]>
instructlab · Jul 25, 2024 · eea8521 · eea8521
1 parent 9fdeb87
commit eea8521
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/requirements-cuda.txt b/requirements-cuda.txt
@@ -1,2 +1,2 @@
 flash-attn>=2.4.0
-bitsandbytes
+bitsandbytes>=0.43.1
diff --git a/requirements.txt b/requirements.txt
@@ -16,7 +16,7 @@ numba
 numpy>=1.23.5,<2.0.0 ; python_version == '3.10'
 numpy>=1.26.4,<2.0.0 ; python_version != '3.10'
 rich
-instructlab-dolomite
+instructlab-dolomite>=0.1.1
 trl>=0.9.4
 peft
 pydantic>=2.7.0