Update pyproject.toml

2024-01-27 17:38:30 +11:00 · 2024-01-27 17:38:30 +11:00 · 8ed03f5f45
commit 8ed03f5f45
parent af65cb0d3d
1 changed files with 2 additions and 2 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@ -36,9 +36,9 @@ huggingface = [
    "transformers>=4.37.0",
    "datasets",
    "sentencepiece",
-    "accelerate",
+    "accelerate>=0.26.1",
    "trl>=0.7.9",
-    "peft",
+    "peft>=0.7.1",
    "tqdm",
    "psutil",
 ]