Poor generation quality when porting fine-tuned BitVAE weights (.ckpt) to Infinity (.pth) #4
Description
When I finetuned this VAE, I had the problem to use it in the 'Infinity' like the image shown.
I found you used ckpt to save the paprameters in this 'BitVAE'. However, in the 'Infinity', you used pth format file like infinity_vae_d32.pth to save it. I extracted the same key and values from BitVAE ckpt file to create a new 'my_model.pth' file. The performance of generated images according to 'my_model.pth' file is very bad. Do you know how to solve this problem?
