Krea 2 Turbo 2 步蒸馏 LoRA:两步即可出草稿

ComfyUI Wikinews

一款社区 DMD2 蒸馏 LoRA 可在 cfg 1.0 下以两步运行 Krea 2 Turbo,快速产出预览与草稿,并附带一套仅使用原生节点的即用型 ComfyUI 工作流。

Krea 2 Turbo 4 步蒸馏 LoRA 的作者又带来了更激进的压缩版本:Krea2-Turbo-Distill-2step-LoRA 将 Krea 2 Turbo 从 8 步降到 2 步,去噪速度约提升 4.2 倍(1024x1024 下从 81.4 秒降至 19.5 秒)。它被定位为预览与草稿工具,最终成图仍推荐使用 4 步 LoRA。

Turbo 8 步Turbo 2 步原版加 LoRA 的 2 步
Turbo 8 步Turbo 2 步:空洞而发虚加 LoRA 的 2 步:细节恢复

分布匹配,而非模仿

与 4 步版本一样,该适配器是一个普通的 rank-64 LoRA,采用分布匹配(DMD2)训练:学生模型学习匹配教师模型可能生成的输出,而不是复制它的精确轨迹,从而避免两步采样因平均而变得模糊、出现重影边缘。学生模型在 sigmas 1.0 和 0.7595 处进行两次调用,也就是教师 8 步网格中的第 1 个和第 5 个点,其间使用原生 Euler 采样器。

当前模型(chk00017464,仍标记为训练进行中)针对的是更早版本遗留的问题:大尺寸下的颗粒感与网格图案、小尺寸人脸以及密集细节。在 12 个训练分辨率上,细纹理能量测得为 8 步教师模型的 1.01 至 1.17 倍,盲测的提示词遵循度评分达到 240 分中的 239 分。训练在单张 RTX 3090 上完成。

ComfyUI 使用方法

该仓库提供了采用 ComfyUI 键名的 krea2_turbo_2step_rank_64_lora_comfyui.safetensors,以及 krea2_turbo_2step_lora_t2i.json,这是一个完全由原生节点搭建的即用工作流。文件名永不改变:更新的模型会原地替换这两个文件。

ComfyUI 工作流图

随附的工作流:Krea 2 Turbo 加 2 步 LoRA,仅使用原生节点。

设置:步数 2cfg 1.0,采样器 euler / simple,LoRA 强度 1.0。有一个值得反复提醒的坑:ComfyUI 用 1.0 表示“不使用 CFG”,而不是 diffusers 中的 0.0。在 ComfyUI 中填 0.0 是另一回事,不会得到训练时的行为表现。

获取方式

评论

使用 GitHub 登录后即可参与讨论。

评论加载中…
Krea 2 Turbo 2 步蒸馏 LoRA:两步即可出草稿 | ComfyUI Wiki