Skip to content

Commit

Permalink
chore(training): add TRL and PEFT as dependencies
Browse files Browse the repository at this point in the history
Peft was already in the AMI, but hte version was not fixed, and TRL was
not installed, but it is required for the SFT LoRA fine tune LLM
tutorial.
  • Loading branch information
tengomucho committed Jan 28, 2025
1 parent a005c5c commit 81dd858
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,8 @@
"huggingface_hub >= 0.20.1",
"numpy>=1.22.2, <=1.25.2",
"protobuf>=3.20.3, <4",
"trl==0.11.4",
"peft"=="0.14.0",
]

TESTS_REQUIRE = [
Expand All @@ -32,8 +34,6 @@
"diffusers>=0.28.0, <=0.30.3",
"safetensors",
"sentence-transformers >= 2.2.0",
"peft",
"trl==0.11.4",
"compel",
"rjieba",
"soundfile",
Expand Down

0 comments on commit 81dd858

Please sign in to comment.