issues
search
cloneofsimo
/
lora
Using Low-rank adaptation to quickly fine-tune diffusion models.
https://arxiv.org/abs/2106.09685
Apache License 2.0
7k
stars
480
forks
source link
Xander updates
#182
Closed
aiXander
closed
1 year ago
aiXander
commented
1 year ago
Several updates across the board:
added a preview function that saves training images to disk so you can see the augmentations
plot the loss curve during training
create a "person" template on top of "object" and "style" with better prompts
make the face crop region 25% larger to get the hairstyle included in the lora
scale the h_flip prob from 50% to 25% at end of training (maybe disable this for objects?)
save a dict of all the args used to train a LoRA (helps to compare settings)
print the number of trainable lora params
Several updates across the board: