Spaces:
Runtime error
Runtime error
user
commited on
Commit
•
da1148c
1
Parent(s):
013a220
Added xformers
Browse files- requirements.txt +2 -1
- trainer.py +1 -0
requirements.txt
CHANGED
|
@@ -9,4 +9,5 @@ tqdm
|
|
| 9 |
ftfy
|
| 10 |
Pillow
|
| 11 |
bitsandbytes
|
| 12 |
-
gradio
|
|
|
|
|
|
| 9 |
ftfy
|
| 10 |
Pillow
|
| 11 |
bitsandbytes
|
| 12 |
+
gradio
|
| 13 |
+
xformers
|
trainer.py
CHANGED
|
@@ -115,6 +115,7 @@ class Trainer:
|
|
| 115 |
--gradient_accumulation_steps={gradient_accumulation} \
|
| 116 |
--learning_rate={learning_rate} \
|
| 117 |
--max_train_steps={n_steps} \
|
|
|
|
| 118 |
--train_batch_size=1 \
|
| 119 |
--lr_scheduler=constant \
|
| 120 |
--lr_warmup_steps=0 \
|
|
|
|
| 115 |
--gradient_accumulation_steps={gradient_accumulation} \
|
| 116 |
--learning_rate={learning_rate} \
|
| 117 |
--max_train_steps={n_steps} \
|
| 118 |
+
--enable_xformers_memory_efficient_attention \
|
| 119 |
--train_batch_size=1 \
|
| 120 |
--lr_scheduler=constant \
|
| 121 |
--lr_warmup_steps=0 \
|