New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open

zwt811 opened this issue Sep 25, 2024 · 1 comment

zwt811 commented Sep 25, 2024

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

Owner

zsyOAOA commented Sep 25, 2024

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment