Training on whole unet or just lora

[Tianhang]

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!

[Xiao Fu]

Hi, we trained the entire unet.