Warlord-K commited on
Commit
bd9d13c
1 Parent(s): 94720b9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -4
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
- - **[Fine-Tune](https://github.com/huggingface/diffusers/blob/main/examples/text_to_image/train_text_to_image_sdxl.py):**
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-sdxl" \
170
  --validation_prompt="cute dragon creature" --report_to="wandb" \
171
  --push_to_hub
172
  ```
173
- - **[LoRA](https://github.com/huggingface/diffusers/blob/main/examples/text_to_image/train_text_to_image_lora_sdxl.py):**
 
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="sdxl-pokemon-model" \
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):**