下载的stable diffudion 模子如何转换到diffusers可用的格式 ...

1.背景
从liblibai或者civitai上下载的模子，通常是safetensors或者cpkt格式的模子。而在diffusers库中，常用的是.bin并搭配json配置文件。两者不能直接使用。
各种格式的区别参考大模子中 .safetensors 文件、.ckpt文件和.pth以及.bin文件区别、加载和保存以及转换方式_safetensors文件-CSDN博客
2、方法
在diffusers中提供了转换工具，具体位置在diffusers库中 ./scripts/convert_original_stable_diffusion_to_diffusers.py。
对于safetensors，使用命令：

1. python ./scripts/convert_original_stable_diffusion_to_diffusers.py
2. --checkpoint_path xxx.safetensors
3. --dump_path save_dir
4. --from_safetensors

复制代码

对于cpkt，使用命令：

1. python ./scripts/convert_original_stable_diffusion_to_diffusers.py
2. --checkpoint_path xxx.ckpt
3. --dump_path save_dir

复制代码

3、模子使用
（1）假如使用整个模子：

1. pipe = StableDiffusionPipeline.from_pretrained("模型保存位置",  torch_dtype=torch.float16)

复制代码

（2）假如只想使用unet模子：

1. unet = UNet.from_pretrained(“模型保存位置”, subfolder="unet").to(dtype=weight_dtype, device="cuda")
2.    

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。