7 Commits

Author	SHA1	Message	Date
ykume	0dfffcd88a	remove unnecessary import	2023-06-11 21:46:05 +09:00
ykume	9e1683cf2b	support sdpa	2023-06-11 21:26:15 +09:00
ykume	035dd3a900	fix mem_eff_attn does not work	2023-06-11 17:08:21 +09:00
ykume	7f6b581ef8	support memory efficient attn (not xformers)	2023-06-11 16:54:41 +09:00
Kohya S	dccdb8771c	support sample generation in training	2023-06-07 08:12:52 +09:00
Kohya S	c0a7df9ee1	fix eps value, enable xformers, etc.	2023-06-03 21:29:27 +09:00
Kohya S	5db792b10b	initial commit for original U-Net	2023-06-03 19:24:47 +09:00