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):**
|