31 commits

Author	SHA1	Message	Date
Shen-Chenhui	48f027f504	longer training config	2024-04-21 10:29:24 +08:00
Shen-Chenhui	679ca934a2	config pipeline	2024-04-21 10:26:30 +08:00
Shen-Chenhui	19187b70f5	allow specifying outputs in command line	2024-04-19 15:56:04 +08:00
Shen-Chenhui	ef18b2e6d8	debug	2024-04-16 18:20:19 +08:00
Shen-Chenhui	becff9d86f	inference v2	2024-04-16 18:15:12 +08:00
Shen-Chenhui	a4025b0ea7	magvit v2 enc dec arc	2024-04-09 17:49:01 +08:00
Shen-Chenhui	ae6bffb8c5	restore train vae script	2024-04-09 09:34:59 +08:00
Shen-Chenhui	d6c0131aab	debug	2024-04-08 17:19:48 +08:00
Shen-Chenhui	a634f7327e	debug	2024-04-08 17:17:22 +08:00
Shen-Chenhui	a401630e7a	debug	2024-04-08 17:03:27 +08:00
Shen-Chenhui	c6181ebdfd	debug	2024-04-08 16:36:54 +08:00
Shen-Chenhui	e785961ca5	debug	2024-04-08 16:29:17 +08:00
Shen-Chenhui	11764f2b42	debug	2024-04-08 16:10:08 +08:00
Shen-Chenhui	4b27448a49	debug	2024-04-08 14:22:21 +08:00
Shen-Chenhui	4c3686c7a8	debug	2024-04-08 10:56:12 +08:00
Shen-Chenhui	1b66919091	debug	2024-04-08 10:53:44 +08:00
Shen-Chenhui	44b8142b4f	add wrapper for pretrained loading	2024-04-05 16:01:24 +08:00
Shen-Chenhui	7624ec6f49	debug	2024-04-04 15:33:39 +08:00
Shen-Chenhui	d08d52ea03	config add is_vae	2024-04-04 15:23:57 +08:00
Shen-Chenhui	e9776584e4	allow vae not to have prompt file	2024-04-04 15:06:36 +08:00
Shen-Chenhui	9dc9afb254	add data path to inference config	2024-04-04 14:39:04 +08:00
Shen-Chenhui	dfdf8db491	add inference config ... add readme complete inference script TODO: DEBUG inference	2024-04-04 14:11:25 +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	ebb3dc4d59	enable wandb	2024-04-02 11:47:56 +08:00
Shen-Chenhui	6bed1bdd0b	working perceptual loss	2024-04-01 16:11:21 +08:00
Shen-Chenhui	3d6d499543	add perceptual loss	2024-04-01 13:57:25 +08:00
Shen-Chenhui	d1e57cbfd5	debug	2024-03-29 10:30:56 +08:00
shenchenhui	71b5c6e805	debug	2024-03-28 17:40:48 +08:00
shenchenhui	73ec879f71	debug	2024-03-28 16:13:50 +08:00
shenchenhui	3d9addba17	debug	2024-03-28 16:09:43 +08:00
shenchenhui	a5917c6a3e	added vae3d training code	2024-03-28 15:12:20 +08:00