ostris / ai-toolkit

Various AI scripts. Mostly Stable Diffusion stuff.
MIT License
2.76k stars 265 forks source link

distorted faces. What am i doing wrong? it dosnt learn tthe face #82

Open protector131090 opened 1 month ago

protector131090 commented 1 month ago

1723823891168__000001500_0 1723661539532__000003750_0 1723788354985__000004250_0

Why am i getting weird distortion in faces? im using 24 vram config. i tried on multiple people datasets (Xl works amazing with them) but Flux dosnt learn them. it distorts faces and overtrains.

protector131090 commented 1 month ago

job: extension config: name: OPA_RANK64 process:

WarAnakin commented 1 month ago

oh wow, that's a strange one. Only thing i can think of is changing the training type to content, see if that works

davsharian commented 1 month ago

it's probably overcooked, try less steps, 1500-2500 for datset of 15-20 images.

martintomov commented 1 month ago

did you solve this?

try:


linear: 32 # or 16
linear_alpha: 32 # or 16
steps: 2000
linear_timesteps: true
jaretburkett commented 1 month ago

Are you using webp images? Someone reported a bug with webp images previously. They are currently not officially supported.

protector131090 commented 4 weeks ago

did you solve this?

try:

linear: 32 # or 16
linear_alpha: 32 # or 16
steps: 2000
linear_timesteps: true

yes. Lower the rank the better. Rank 16 with same dataset is perfect. 1024 only is even better