Update README.md
Browse files
README.md
CHANGED
@@ -150,7 +150,7 @@ The Segmind Stable Diffusion Model is suitable for research and practical applic
|
|
150 |
|
151 |
The Segmind Stable Diffusion Model can also be used directly with the 🧨 Diffusers library training scripts for further training, including:
|
152 |
|
153 |
-
- **[
|
154 |
```bash
|
155 |
export MODEL_NAME="segmind/SSD-1B"
|
156 |
export VAE_NAME="madebyollin/sdxl-vae-fp16-fix"
|
@@ -166,11 +166,12 @@ accelerate launch train_text_to_image_lora_sdxl.py \
|
|
166 |
--learning_rate=1e-04 --lr_scheduler="constant" --lr_warmup_steps=0 \
|
167 |
--mixed_precision="fp16" \
|
168 |
--seed=42 \
|
169 |
-
--output_dir="sd-pokemon-model-lora-
|
170 |
--validation_prompt="cute dragon creature" --report_to="wandb" \
|
171 |
--push_to_hub
|
172 |
```
|
173 |
-
|
|
|
174 |
```bash
|
175 |
export MODEL_NAME="segmind/SSD-1B"
|
176 |
export VAE_NAME="madebyollin/sdxl-vae-fp16-fix"
|
@@ -192,7 +193,7 @@ accelerate launch train_text_to_image_sdxl.py \
|
|
192 |
--report_to="wandb" \
|
193 |
--validation_prompt="a cute Sundar Pichai creature" --validation_epochs 5 \
|
194 |
--checkpointing_steps=5000 \
|
195 |
-
--output_dir="
|
196 |
--push_to_hub
|
197 |
```
|
198 |
- **[Dreambooth LoRA](https://github.com/huggingface/diffusers/blob/main/examples/dreambooth/train_dreambooth_lora_sdxl.py):**
|
|
|
150 |
|
151 |
The Segmind Stable Diffusion Model can also be used directly with the 🧨 Diffusers library training scripts for further training, including:
|
152 |
|
153 |
+
- **[LoRA](https://github.com/huggingface/diffusers/blob/main/examples/text_to_image/train_text_to_image_lora_sdxl.py):**
|
154 |
```bash
|
155 |
export MODEL_NAME="segmind/SSD-1B"
|
156 |
export VAE_NAME="madebyollin/sdxl-vae-fp16-fix"
|
|
|
166 |
--learning_rate=1e-04 --lr_scheduler="constant" --lr_warmup_steps=0 \
|
167 |
--mixed_precision="fp16" \
|
168 |
--seed=42 \
|
169 |
+
--output_dir="sd-pokemon-model-lora-ssd" \
|
170 |
--validation_prompt="cute dragon creature" --report_to="wandb" \
|
171 |
--push_to_hub
|
172 |
```
|
173 |
+
|
174 |
+
- **[Fine-Tune](https://github.com/huggingface/diffusers/blob/main/examples/text_to_image/train_text_to_image_sdxl.py):**
|
175 |
```bash
|
176 |
export MODEL_NAME="segmind/SSD-1B"
|
177 |
export VAE_NAME="madebyollin/sdxl-vae-fp16-fix"
|
|
|
193 |
--report_to="wandb" \
|
194 |
--validation_prompt="a cute Sundar Pichai creature" --validation_epochs 5 \
|
195 |
--checkpointing_steps=5000 \
|
196 |
+
--output_dir="ssd-pokemon-model" \
|
197 |
--push_to_hub
|
198 |
```
|
199 |
- **[Dreambooth LoRA](https://github.com/huggingface/diffusers/blob/main/examples/dreambooth/train_dreambooth_lora_sdxl.py):**
|