本文主要是介绍模型格式转化diffusion_pytorch_model.safetensors转为diffusion_pytorch_model.bin,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
问题描述
SError: Error no file named config.json found in directory /share/huanggao/zjc/code_mid/model_yw/bk-base-2m/unet.
无法读取unet
下载模型
我们从网上下载的模型
通过
pipe = StableDiffusionPipeline.from_pretrained("nota-ai/bk-sdm-base-2m", torch_dtype=torch.float16)指令下载的unet是
diffusion_pytorch_model.safetensors格式的
我从网上直接下载的模型路径
/root/.cache/huggingface/hub/models--nota-ai--bk-sdm-base-2m/snapshots/e8b5597155c5b2c77585570b99113f1c77b97338/unet
转移到
/share/huanggao/zjc/code_mid/model_yw/bk-base-2m/unet
文件格式
config.json
diffusion_pytorch_model.safetensors
smallsd路径
~/.cache/huggingface/hub/models--OFA-Sys--small-stable-diffusion-v0/snapshots/38e10e5e71e8fbf717a47a81e7543cd01c1a8140/unet
格式是(diffusers格式)
文件格式
config.json
diffusion_pytorch_model.bin
转化文件路径
python ./scripts/convert_original_stable_diffusion_to_diffusers.py
转化文件
convert_original_stable_diffusion_to_diffusers.py
转化方法
python ./scripts/convert_original_stable_diffusion_to_diffusers.py
--checkpoint_path xxx.safetensors
--dump_path save_dir
--from_safetensors
python ./convert_original_stable_diffusion_to_diffusers.py --checkpoint_path /root/.cache/huggingface/hub/models--nota-ai--bk-sdm-base-2m/snapshots/e8b5597155c5b2c77585570b99113f1c77b97338/unet/diffusion_pytorch_model.safetensors --dump_path /share/huanggao/zjc/code_mid --from_safetensors
转化过程会报错
KeyError: 'time_embed.0.weight'
这篇关于模型格式转化diffusion_pytorch_model.safetensors转为diffusion_pytorch_model.bin的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!