Example dreambooth lora script

PixArt-alpha / PixArt-sigma

PixArt-Σ: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation

https://pixart-alpha.github.io/PixArt-sigma-project/

GNU Affero General Public License v3.0

1.44k stars 67 forks source link

Example dreambooth lora script #128

Closed frutiemax92 closed 7 hours ago

frutiemax92 commented 2 days ago

Example command: python .\train_scripts\train_dreambooth_lora.py --pretrained_model_path "PixArt-alpha/PixArt-Sigma-XL-2-1024-MS" --images_folder "Test" --captions_folder "TestCaps" --class_id "cat" --unique_id "h7g3" --output_folder "Test" --validation_prompt "h7g3 cat" --transformer_output_path "TestCheckpoint" --epochs_per_save 1 --num_regularization_passes 1 --transformer_output_path TestCheckpoint --merge_path Test4