diffusers
2d1f2182 - example: Train Instruct pix2 pix with lora implementation (#6469)

Commit
1 year ago
example: Train Instruct pix2 pix with lora implementation (#6469) * base template file - train_instruct_pix2pix.py * additional import and parser argument requried for lora * finetune only instructpix2pix model -- no need to include these layers * inject lora layers * freeze unet model -- only lora layers are trained * training modifications to train only lora parameters * store only lora parameters * move train script to research project * run quality and style code checks * move train script to a new folder * add README * update README * update references in README --------- Co-authored-by: Rahul Raman <rahulraman@gmail.com> Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
Author
Parents
Loading