김정현
commited on
Commit
·
50f13f7
1
Parent(s):
4fd7beb
test
Browse files- requirements.txt +6 -4
requirements.txt
CHANGED
|
@@ -1,9 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
tokenizers==0.15.2
|
| 2 |
torch==2.2.1+cu118
|
| 3 |
torchaudio==2.2.1+cu118
|
| 4 |
torchvision==0.17.1+cu118
|
| 5 |
transformers==4.38.1
|
| 6 |
-
|
| 7 |
-
peft
|
| 8 |
-
sentencepiece
|
| 9 |
-
accelerate
|
|
|
|
| 1 |
+
accelerate @ git+https://github.com/huggingface/accelerate.git@97d2168e5953fe7373a06c69c02c5a00a84d5344
|
| 2 |
+
bitsandbytes==0.42.0
|
| 3 |
+
datasets==2.17.1
|
| 4 |
+
huggingface-hub==0.20.3
|
| 5 |
+
peft==0.8.2
|
| 6 |
tokenizers==0.15.2
|
| 7 |
torch==2.2.1+cu118
|
| 8 |
torchaudio==2.2.1+cu118
|
| 9 |
torchvision==0.17.1+cu118
|
| 10 |
transformers==4.38.1
|
| 11 |
+
trl==0.7.11
|
|
|
|
|
|
|
|
|