search

haoningwu3639 / SimpleSDXL

A simple and flexible PyTorch implementation of StableDiffusion-XL based on diffusers.
12 stars 1 forks source link

支持sdxl-lora训练 #2

Closed JoshonSmith closed 4 weeks ago

JoshonSmith commented 3 months ago

支持lora训练模式吗

haoningwu3639 commented 3 months ago

Well, it should support lora training, but I've only implemented the basic finetuning functionality. Specifically, you can refer to the implementation of lora layers in the library, add it to unet, freeze the other parameters, and train only the lora layer.