KohakuBlueleaf/LyCORIS

Lora beYond Conventional methods, Other Rank adaptation Implementations for Stable diffusion.

解决的问题

LyCORIS 提供了一套参数高效的微调(PEFT)算法,用于定制 Stable Diffusion 模型。它允许用户训练小型、可移植的适配器模型(而非完整的模型检查点),这些模型可以轻松共享和组合,从而显著降低微调所需的存储和计算资源。

工作原理

该项目实现了多种超越标准 LoRA 的低秩适配技术。它提供独立的 PyTorch 包装器,可应用于任何模块,同时也支持与流行的训练脚本 sd-scripts 以及 sd-webui、ComfyUI 等界面的集成。

实现的关键算法包括:

  • LoRA/LoCon:标准低秩适配及其在卷积层的扩展
  • LoHa:低秩哈达玛积
  • LoKr:克罗内克积适配
  • (IA)³:注入式适配器
  • DyLoRA:动态 LoRA

最近的更新引入了使用 Triton 和 TileLang 的实验性融合内核,显著减少了训练和推理过程中的设备时间和显存占用。

适用人群

  • AI 艺术家和模型训练者:希望在不训练完整模型的情况下,为特定角色、风格或概念定制 Stable Diffusion 的用户
  • 机器学习研究人员:探索图像生成中不同低秩适配方法的研究者
  • 开发者:需要集成多个微调适配器的图像生成流水线构建者

主要亮点

  • 丰富的算法支持:实现多种 PEFT 方法,包括 LoHa、LoKr 和 (IA)³
  • 高性能:实验性融合内核(Triton/TileLang)相比 eager 模式可实现 1.4x–7.3x 的加速
  • 广泛的集成:原生支持 sd-webui、ComfyUI 和 InvokeAI
  • 灵活的 API:可作为任意 PyTorch 模块的独立包装器使用,也可通过高级训练脚本调用

相关

  • Dispatch
  • 项目
  • 项目
  • Dispatch
  • 项目