Robin Rombach 4c8bff393a Merge pull request #51 from crowsonkb/main ... Add PLMS sampling and do one 2x size batch per sampling step		2022-04-15 17:35:51 +02:00
..
data	add code	2021-12-21 03:23:41 +01:00
models	Combine the two classifier-free guidance model outputs into a single batch	2022-04-05 11:35:05 -07:00
modules	do not fix num_heads to one in legacy mode	2022-04-15 17:24:11 +02:00
lr_scheduler.py	add code	2021-12-21 03:23:41 +01:00
util.py	add code	2021-12-21 03:23:41 +01:00