Update README.md
Browse files
README.md
CHANGED
@@ -18,6 +18,30 @@ Training steps included:
|
|
18 |
|
19 |
- train-text-encoder fine tuning
|
20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
### 🧨 Diffusers
|
22 |
|
23 |
This model can be used just like any other Stable Diffusion model. For more information,
|
|
|
18 |
|
19 |
- train-text-encoder fine tuning
|
20 |
|
21 |
+
|
22 |
+
```
|
23 |
+
accelerate launch train_dreambooth.py \
|
24 |
+
--pretrained_model_name_or_path=$MODEL_NAMEi \
|
25 |
+
--instance_data_dir=$INSTANCE_DIR \
|
26 |
+
--class_data_dir=$CLASS_DIR \
|
27 |
+
--output_dir=$OUTPUT_DIR \
|
28 |
+
--instance_prompt="sks style" \
|
29 |
+
--resolution=512 \
|
30 |
+
--train_batch_size=1 \
|
31 |
+
--gradient_accumulation_steps=1 \
|
32 |
+
--gradient_checkpointing \
|
33 |
+
--use_8bit_adam \
|
34 |
+
--enable_xformers_memory_efficient_attention \
|
35 |
+
--set_grads_to_none \
|
36 |
+
--learning_rate=1e-6 \
|
37 |
+
--lr_scheduler="constant" \
|
38 |
+
--lr_warmup_steps=0 \
|
39 |
+
--max_train_steps=2000 \
|
40 |
+
--train_text_encoder \
|
41 |
+
--mixed_precision=fp16
|
42 |
+
```
|
43 |
+
|
44 |
+
|
45 |
### 🧨 Diffusers
|
46 |
|
47 |
This model can be used just like any other Stable Diffusion model. For more information,
|