Commit Graph

9 Commits

Author SHA1 Message Date
harubaru f980137430 add fp16 training 2022-10-15 10:57:00 -07:00
hasuwoof 60c04259fa Implement CLIP extensions (training + inference) 2022-10-14 19:31:53 +01:00
harubaru c4d8e967e9 oops forgot layer norm
this is probably why i was getting garbage in training
2022-10-13 01:39:01 -07:00
harubaru adbbc73e19 fix hidden states 2022-10-12 18:48:22 -07:00
harubaru 4f9070af3c use penultimate layer
from NAI blogpost and https://arxiv.org/pdf/2205.11487.pdf
2022-10-12 17:55:56 -07:00
Robin Rombach 2ff270f4e0 stable diffusion 2022-08-10 16:30:49 +02:00
rromb f13bf9bf46 add vqgan loss with codebook statistic eval 2022-02-21 15:06:50 +01:00
ablattmann 171cf29fb5 add configs for training unconditional/class-conditional ldms 2021-12-22 15:57:23 +01:00
ablattmann e66308c7f2 add code 2021-12-21 03:23:41 +01:00