How to Train/Finetune the ShapeVAE ? #178
Description
Hiya, thanks for your great work. I want to finetune the ShapeVAE on my own dataset, but there are only training scripts for the DiT. Could u provide the training code for the ShapeVAE or give me some advice (e.g., about the hyper params)? Thx!