Do your .ckpt model weight files have /unet/diffusion_pytorch_model.bin in them?

#95
by nbkn865 - opened

I'm getting this error when using your model weight files:

<my-dir>/v1-5-pruned-emaonly.ckpt/unet/diffusion_pytorch_model.bin can not be found., where <my-dir> refers to my work directory, which I've abstracted here since it's not relevant

I'm new to the .ckpt file format, but based on the above, are they some sort of compressed file format that has directories in them, e.g. unet?

This is the code that generates the above error: https://mmediting.readthedocs.io/en/dev-1.x/model_zoo/text2image.html#quick-start

I edited the config file stable-diffusion_ddim_denoisingunet.py as follows, to point to the .ckpt file:

init_cfg = dict(type='Pretrained', pretrained_model_path='<my-dir>/v1-5-pruned-emaonly.ckpt')

Do you know if the issue is with the OpenMMLab package mmedit, or with the .ckpt files?

Sign up or log in to comment