Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[examples] Fix CLI argument in the launch script command for text2image with LoRA (#2171) 

* Update README.md

* Update README.md
This commit is contained in:
Sayak Paul 2023-01-31 14:17:09 +05:30 committed by GitHub
parent cedafb8600
commit 7d96b38b70
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/text_to_image/README.md
View File

@ -148,7 +148,7 @@ huggingface-cli login
Now we can start training! Now we can start training!
```bash ```bash
accelerate --mixed_precision="fp16" launch train_text_to_image_lora.py \ accelerate launch --mixed_precision="fp16" train_text_to_image_lora.py \
--pretrained_model_name_or_path=$MODEL_NAME \ --pretrained_model_name_or_path=$MODEL_NAME \
--dataset_name=$DATASET_NAME --caption_column="text" \ --dataset_name=$DATASET_NAME --caption_column="text" \
--resolution=512 --random_flip \ --resolution=512 --random_flip \
@ -157,7 +157,7 @@ accelerate --mixed_precision="fp16" launch train_text_to_image_lora.py \
--learning_rate=1e-04 --lr_scheduler="constant" --lr_warmup_steps=0 \ --learning_rate=1e-04 --lr_scheduler="constant" --lr_warmup_steps=0 \
--seed=42 \ --seed=42 \
--output_dir="sd-pokemon-model-lora" \ --output_dir="sd-pokemon-model-lora" \
--save_sample_prompt="cute dragon creature" --report_to="wandb" --validation_prompt="cute dragon creature" --report_to="wandb"
``` ```
The above command will also run inference as fine-tuning progresses and log the results to Weights and Biases. The above command will also run inference as fine-tuning progresses and log the results to Weights and Biases.