search

zsyOAOA / ResShift

ResShift: Efficient Diffusion Model for Image Super-resolution by Residual Shifting (NeurIPS@2023 Spotlight, TPAMI@2024)
Other
944 stars 50 forks source link

The model trained on my own dataset gives an error when using inference_resshift.py for inference. #95

Open zwt811 opened 1 month ago

zwt811 commented 1 month ago

I used autoencoder_vq_f4.pth as the VQGAN model to train my own dataset, but during inference, I encountered the error KeyError: 'vae'.

zsyOAOA commented 1 month ago

For the task of deblurring, I finetuned the autoencoder with LoRA, which requires the key "vae". Please carefully read the code of this part.

https://github.com/zsyOAOA/ResShift/blob/journal/sampler.py#L103