Commit Graph

22 Commits

Author SHA1 Message Date
harubaru 76efa00de3 fix deprecation warnings in dataloader 2022-10-15 18:43:28 -07:00
harubaru f980137430 add fp16 training 2022-10-15 10:57:00 -07:00
harubaru 9581fbc226
Merge pull request #24 from hasuwoof/feature/extended_clip
Implement CLIP extensions (training + inference)
2022-10-14 16:27:21 -07:00
hasuwoof 60c04259fa Implement CLIP extensions (training + inference) 2022-10-14 19:31:53 +01:00
Ikko Ashimine 8b0f40809e
Fix typo in ddpm.py
postions -> positions
2022-10-14 12:14:25 +09: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 169b63e179 add unconditional guidance training 2022-10-12 18:28:44 -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
harubaru c5f2775beb Add alternate dataloader 2022-09-22 14:56:27 -07:00
harubaru 2e69358d50 Add random tag order during training 2022-09-21 04:57:17 -07:00
harubaru b080f33115 Add variable aspect ratio during training 2022-09-21 04:47:39 -07:00
harubaru c690d005bc fix cropping 2022-09-20 22:24:04 -07:00
harubaru 01b9440d50 dataloader overhaul 2022-09-20 22:00:32 -07:00
harubaru f7bb9b7bbc fix validation 2022-09-06 23:33:12 -07:00
harubaru dc0d0bd579 add finetuning configs 2022-09-02 21:00:05 -07:00
harubaru e6c1e048ea add local dataset code 2022-08-30 14:06:47 -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
Robin Rombach 2b46bcb98c
Update ddpm.py
clean up no.1
2022-01-17 21:24:19 +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