Yujun-Shi / DragDiffusion

[CVPR2024, Highlight] Official code for DragDiffusion
https://yujun-shi.github.io/projects/dragdiffusion.html
Apache License 2.0
1.13k stars 82 forks source link

possible to retain original lora checkpoint for other StableDiffusionPipeline? #24

Open danieltanhx opened 1 year ago

danieltanhx commented 1 year ago

i noticed that your training DragPipeline class will save checkpoint incompatible with StableDiffusionPipeline,StableDiffusionImg2ImgPipeline and (https://github.com/cloneofsimo/lora/blob/master/scripts/run_img2img.ipynb).

Is there a way for me to make this dragpipeline checkpoint compatible with the original lora checkpoint? it will be difficult to integrate with other tools if the lora structure is diff