44 commits

Author	SHA1	Message	Date
Shen-Chenhui	1bbef18f15	strides debug	2024-04-09 18:14:25 +08:00
Shen-Chenhui	a4025b0ea7	magvit v2 enc dec arc	2024-04-09 17:49:01 +08:00
Shen-Chenhui	88be217a13	debug	2024-04-05 16:25:32 +08:00
Shen-Chenhui	e3584b4e43	debug	2024-04-05 16:20:39 +08:00
Shen-Chenhui	44b8142b4f	add wrapper for pretrained loading	2024-04-05 16:01:24 +08:00
Shen-Chenhui	c7506e2461	adding pretrained vae	2024-04-04 15:54:55 +08:00
Shen-Chenhui	a656421c83	debug from_pretrained in progress	2024-04-04 15:41:15 +08:00
Shen-Chenhui	b6c39873ee	finish discriminator arch ... use einops.rearrange in vae model_utils add get_latent_size in vae_3d inference vae_3d WIP	2024-04-03 10:29:01 +08:00
Shen-Chenhui	aa733aa1d7	debug	2024-04-02 11:21:01 +08:00
Shen-Chenhui	8e49c1575b	debug	2024-03-29 10:38:44 +08:00
Shen-Chenhui	6cb4e83d8a	debug	2024-03-29 10:31:58 +08:00
Shen-Chenhui	d1e57cbfd5	debug	2024-03-29 10:30:56 +08:00
Shen-Chenhui	b4b541756e	debug	2024-03-29 10:28:57 +08:00
Shen-Chenhui	a23b800952	debug	2024-03-29 10:17:49 +08:00
Shen-Chenhui	1edc7c60ed	debug	2024-03-29 10:17:09 +08:00
Shen-Chenhui	c9fac9fa2b	debug	2024-03-29 10:14:37 +08:00
Shen-Chenhui	12100a2c94	debug	2024-03-29 10:12:06 +08:00
Shen-Chenhui	b185c54160	debug	2024-03-29 10:01:39 +08:00
Shen-Chenhui	1f81907ceb	debug	2024-03-29 09:59:07 +08:00
Shen-Chenhui	3aa248d7cf	debug	2024-03-29 09:31:43 +08:00
Shen-Chenhui	907c71bcd6	debug	2024-03-28 23:24:37 +08:00
Shen-Chenhui	1c3d6d4c4c	debug	2024-03-28 23:12:36 +08:00
shenchenhui	d927619833	debug	2024-03-28 23:08:32 +08:00
shenchenhui	21296e1497	debug	2024-03-28 22:59:01 +08:00
shenchenhui	6598eec0b8	debug	2024-03-28 22:53:57 +08:00
shenchenhui	356ff604c0	debug	2024-03-28 22:51:22 +08:00
shenchenhui	f1a6f7523e	debug	2024-03-28 21:41:39 +08:00
shenchenhui	9d3bdcafc4	debug	2024-03-28 21:19:35 +08:00
shenchenhui	41f7b9b01e	debug	2024-03-28 18:17:42 +08:00
shenchenhui	a0f9aa87c9	debug	2024-03-28 18:03:37 +08:00
shenchenhui	7f105cbdfa	debug	2024-03-28 18:02:30 +08:00
shenchenhui	0aec391d80	debug	2024-03-28 17:57:38 +08:00
shenchenhui	5935c5b10b	debug	2024-03-28 17:36:31 +08:00
shenchenhui	f617d40505	debug	2024-03-28 17:26:55 +08:00
shenchenhui	9b75f917ed	debug	2024-03-28 17:25:46 +08:00
shenchenhui	d80ef70020	debug	2024-03-28 17:15:04 +08:00
shenchenhui	bdf8a1d144	debug	2024-03-28 17:13:16 +08:00
shenchenhui	61c7a0d029	debug	2024-03-28 17:09:37 +08:00
shenchenhui	3755547fb9	debug	2024-03-28 17:07:56 +08:00
shenchenhui	04c03eeed4	debug	2024-03-28 16:55:37 +08:00
shenchenhui	64a1e76100	debug	2024-03-28 16:46:38 +08:00
shenchenhui	b3f2dacc69	debug	2024-03-28 16:44:41 +08:00
shenchenhui	8a5de06b44	remove ml_collections	2024-03-28 16:06:00 +08:00
shenchenhui	a5917c6a3e	added vae3d training code	2024-03-28 15:12:20 +08:00