Dear professor, thank you for your contribution! I trained VQGAN on my own dataset and found that there are some differences between the network structure of VQGAN and the original VAGAN structure. After loading the model, an error will be reported. Can you tell me the details of your processing?