当前位置: 首页 > article >正文

下载的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,使用命令:

python ./scripts/convert_original_stable_diffusion_to_diffusers.py
--checkpoint_path xxx.safetensors
--dump_path save_dir
--from_safetensors

对于cpkt,使用命令:

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

3、模型使用

(1)如果使用整个模型:

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

(2)如果只想使用unet模型:

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


http://www.kler.cn/a/502765.html

相关文章:

  • 源码编译安装httpd 2.4,提供系统服务管理脚本并测试(两种方法实现)
  • redis缓存篇知识点总结
  • 73.矩阵置零 python
  • C++ ——— 内部类
  • HBuilderX打包ios保姆式教程
  • python 寻找数据拐点
  • SQLynx 数据库管理平台 3.6.0 全新发布:全面支持华为数据库和ClickHouse,代码提示更智能!
  • 软考信安21~网络设备安全
  • Android Room 构建问题:There are multiple good constructors
  • 备战春招—高频芯片设计面试题
  • DuckDB:星号(*)表达式完整指南
  • HIVE技术
  • 【AscendC】tiling方案设计不当引起的一个时隐时现的bug
  • CNN中模型的参数量与FLOPs计算
  • Spring MVC数据绑定POJO类型
  • 【动态规划-矩阵】6.最大正方形
  • Linux 子系统 Ubuntu 安装MySQL 8
  • 【Apache Paimon】-- 为什么选择将 Spark 与 Paimon 集成,解决什么问题?
  • 国产linux系统(银河麒麟,统信uos)使用 PageOffice 实现后台生成单个PDF文档
  • 虚假星标:GitHub上的“刷星”乱象与应对之道
  • 如何解决HTML和CSS相关情况下会导致页面布局不稳定?
  • ImportError: attempted relative import with no known parent package 报错的解决!
  • 2025年,华为认证HCIA、HCIP、HCIE 该如何选择?
  • 任务调度系统Quartz.net详解1-基本流程及Core表达式
  • 验证码的设置
  • Linux离线部署ELK