229 commits

Author	SHA1	Message	Date
zhengzangw	f77370fff6	[feat] vae micro frame	2024-05-05 15:25:17 +00:00
Zangwei Zheng	c4f17b41fc	[exp] update config	2024-05-05 20:26:26 +08:00
Zheng Zangwei (Alex Zheng)	f941fa95dc	Merge branch 'dev/v1.2' into vae-clean-new	2024-05-04 20:26:42 +08:00
Zangwei Zheng	d07003b0cd	[fix] rf inference scale	2024-05-04 20:23:32 +08:00
Shen-Chenhui	d8fab103f4	added disc	2024-05-02 09:48:30 +00:00
Shen-Chenhui	bcb997502e	enable disc	2024-05-02 17:28:45 +08:00
Shen-Chenhui	b35b933841	add disc	2024-05-02 16:32:19 +08:00
Zangwei Zheng	00e21e9fed	[wip] adapt multi-res	2024-05-02 15:27:03 +08:00
zhengzangw	02904b4f29	updated	2024-05-02 03:46:01 +00:00
zhengzangw	f7a4fcfc3f	updated	2024-04-30 09:46:27 +00:00
zhengzangw	b89c072eb5	[fix] micro batch size not for training	2024-04-30 09:20:08 +00:00
zhengzangw	ace37cf7c7	refactor	2024-04-30 08:13:20 +00:00
zhengzangw	41e276f5ef	updated	2024-04-30 06:19:40 +00:00
周钰坤	bf633f3e68	Update pixart.py	2024-04-30 13:53:17 +08:00
周钰坤	34711b7bd2	update	2024-04-30 13:37:25 +08:00
周钰坤	59591d8afc	update	2024-04-30 13:36:41 +08:00
zhengzangw	b2f302b379	[wip] try vae	2024-04-29 17:02:22 +00:00
Shen-Chenhui	e3b84247c1	disable t-conv for t=1	2024-04-29 20:06:18 +08:00
Shen-Chenhui	c59b499085	Merge branch 'vae-clean-new' of https://github.com/hpcaitech/Open-Sora-dev into vae-clean-new	2024-04-29 18:10:03 +08:00
Shen-Chenhui	c631358bbf	add z nll loss	2024-04-29 18:09:42 +08:00
zhengzangw	9632b5bcf8	merged	2024-04-29 10:07:09 +00:00
zhengzangw	5c8139521a	update	2024-04-29 10:02:42 +00:00
Shen-Chenhui	b78b469420	fix merge	2024-04-29 17:58:59 +08:00
Shen-Chenhui	f81d8648cf	move out losses to different file	2024-04-29 17:53:03 +08:00
zhengzangw	a2dc903863	update	2024-04-29 09:41:39 +00:00
Shen-Chenhui	4d69671663	clean up code	2024-04-29 17:18:45 +08:00
Shen-Chenhui	98de78910d	config	2024-04-29 15:21:54 +08:00
Shen-Chenhui	7811b8e99a	add Nlayer 3D discriminator	2024-04-29 14:18:57 +08:00
Shen-Chenhui	069ffcc4e3	use 3 layer disc	2024-04-29 14:03:18 +08:00
zhengzangw	f3dcb4d6fb	complete 3b backbone	2024-04-29 06:00:14 +00:00
Shen-Chenhui	8b465b04c5	add vae subfolder support	2024-04-29 11:32:39 +08:00
zhengzangw	84465c0e44	add config for PixArt 1.6B	2024-04-27 13:41:21 +00:00
zhengzangw	7b08fb289c	[fix] pixart-sigma	2024-04-27 13:19:00 +00:00
zhengzangw	d6a16a858e	[wip] multiple resolution	2024-04-27 11:43:15 +00:00
zhengzangw	f551321e49	Merge branch 'main' into dev/v1.2	2024-04-27 11:40:56 +00:00
Shen-Chenhui	c9785342c2	add evals	2024-04-27 16:14:11 +08:00
Zheng Zangwei (Alex Zheng)	8f109b7117	fix qknorm bug (#80)	2024-04-27 15:52:17 +08:00
YuKun Zhou	fddd08243b	Update ckpt_utils.py (#79) ... * Update ckpt_utils.py * update * Update 1x256x256.py * update * update * Update blocks.py * update * Update 1x2048x2048.py	2024-04-27 15:43:50 +08:00
Shen Chenhui	3e7ba9a46d	use modulelist	2024-04-27 14:02:20 +08:00
zhengzangw	478b585024	[wip] debug vae	2024-04-26 07:27:26 +00:00
zhengzangw	a4652a8aef	Merge branch 'main' into vae	2024-04-26 06:56:22 +00:00
zhengzangw	7fcf0c6bfd	fix qknorm bug	2024-04-26 03:59:50 +00:00
Shen-Chenhui	576b44d98e	add lecam storage support	2024-04-26 10:34:00 +08:00
zhengzangw	44e618bbbc	a bunch of update	2024-04-24 09:23:24 +00:00
Shen-Chenhui	9683ec66c7	inference script update	2024-04-24 17:04:53 +08:00
Shen-Chenhui	0138be6e21	grad penalty expr	2024-04-24 14:52:47 +08:00
Shen-Chenhui	988bc3bb65	config lc gp	2024-04-24 10:03:58 +08:00
Shen-Chenhui	48e3a84ba6	enable r1 gradient penalty	2024-04-23 17:51:50 +08:00
Shen-Chenhui	d565477680	disable non-master print	2024-04-23 14:43:02 +08:00
Shen-Chenhui	d0d7da36ea	silence lpips	2024-04-23 11:44:33 +08:00