Training on whole unet or just lora
Tianhang-Cheng opened this issue · 1 comments
Tianhang-Cheng commented
Thanks for the great work!! I am curious whether you only trained the attention layer and lora in unet, or trained the entire unet after loading the pretrained SD weight? Thank you very much!
fuxiao0719 commented
Hi, we trained the entire unet.