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