harubaru / waifu-diffusion

stable diffusion finetuned on weeb stuff
GNU Affero General Public License v3.0
1.94k stars 177 forks source link

Add (vague) V-Objective option #51

Closed chavinlo closed 1 year ago

chavinlo commented 1 year ago

There are some things I think could be done better (I will if I have time)

the problem with this is that during inference it gets ignored for some reason and goes directly with the normal (non vobjective) DDIMScheduler. idk if it comes from the config check, so I'm just forcing it here

currently there are two schedulers that support velocity: ddpm and ddim at least on the diffusers repo

the v objective was added 6h ago, and last release was 3 days ago

chavinlo commented 1 year ago

Do not add yet. Testing it right now and it seems the loss is over the sky

chavinlo commented 1 year ago

Closed as this has been added