Skip to content

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

Description

@alchemist-zsh

你好,训练运行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.

Activity

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions