liang.zhao
commited on
Commit
·
97f0ced
1
Parent(s):
ab1dbe6
update model and config
Browse files- requirements.txt +1 -2
requirements.txt
CHANGED
|
@@ -1,6 +1,5 @@
|
|
| 1 |
tokenizers==0.14.0
|
| 2 |
transformers==4.34.0
|
| 3 |
-
flash_attn==2.0.4
|
| 4 |
torch==2.1.0
|
| 5 |
peft==0.5.0
|
| 6 |
-
|
|
|
|
| 1 |
tokenizers==0.14.0
|
| 2 |
transformers==4.34.0
|
|
|
|
| 3 |
torch==2.1.0
|
| 4 |
peft==0.5.0
|
| 5 |
+
datasets==2.14.1
|