Skip to content
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.

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

训练运行train.py 运行到 vae = AutoencoderKL.from_pretrained(opt.stable_dif_path, subfolder="vae")报错 #22

Closed
alchemist-zsh opened this issue Oct 29, 2024 · 2 comments

Comments

@alchemist-zsh
Copy link

你好,训练运行train.py 运行到 vae = AutoencoderKL.from_pretrained(opt.stable_dif_path, subfolder="vae")报错(如下),没有找到diffusion_pytorch_model.bin,请问能否把需要的runwayml/stable-diffusion-v1-5里面的模型也上传一下,因为huggingface 里面的runwayml/stable-diffusion-v1-5已经删掉了,在其他地方下载的不正确报了这个错误。

Exception has occurred: OSError
Error no file named diffusion_pytorch_model.bin found in directory /media/hzs/data/hzs/One-DM-main/runwayml/stable-diffusion-v1-5.
File "/media/One-DM-main/train.py", line 92, in main
vae = AutoencoderKL.from_pretrained(opt.stable_dif_path, subfolder="vae")
File "/media/One-DM-main/train.py", line 115, in
main(opt)
OSError: Error no file named diffusion_pytorch_model.bin found in directory /media/hzs/data/hzs/One-DM-main/runwayml/stable-diffusion-v1-5.

@dailenson
Copy link
Owner

你好~感谢提醒。stable-diffusion-v1-5已经上传:https://wisemodel.cn/models/SCUT-MMPR/One-DM/file

@dailenson dailenson pinned this issue Oct 29, 2024
@alchemist-zsh
Copy link
Author

可以训练了,谢谢

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants