Commit
•
603eaa1
1
Parent(s):
329ea8d
Update requirements.txt
Browse files- requirements.txt +1 -8
requirements.txt
CHANGED
|
@@ -2,11 +2,4 @@ gradio
|
|
| 2 |
torch
|
| 3 |
transformers
|
| 4 |
peft
|
| 5 |
-
trl
|
| 6 |
-
AutoModelForCausalLM
|
| 7 |
-
AutoTokenizer
|
| 8 |
-
BitsAndBytesConfig
|
| 9 |
-
HfArgumentParser
|
| 10 |
-
TrainingArguments
|
| 11 |
-
pipeline
|
| 12 |
-
logging
|
|
|
|
| 2 |
torch
|
| 3 |
transformers
|
| 4 |
peft
|
| 5 |
+
trl
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|