grafting

像嫁接植物一样组合扩散模型架构,快速找到最优设计。

开源 🏷️完全免费 ✅ 目前可打开 图像生成 🍴 复刻 4 🐛 待处理 1 👀 关注 1
访问官网 ↗ GitHub ↗
GitHub 星标 ★ 77
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源代码,Apache-2.0许可,官网提供在线演示,完全免费。

项目预览

grafting 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类图像生成
开发团队keshik6
定价模式🏷️完全免费
价格说明开源代码,Apache-2.0许可,官网提供在线演示,完全免费。
访问状态✅ 目前可打开
是否开源是
开源协议Apache-2.0
主要语言Jupyter Notebook
技术栈/模型architecture-editing,architecture-research,convolutions,diffusion-models,diffusion-transformer,diffusion-transformers,grafting,hyena-operator,hyena-x,hyena-y,image-generation,linear-attention,mamba-2,model-architecture-editing,model-grafting,post-training,self-attention,sub-quadratic-attention,text-to-image-generation
GitHub 星标★ 77
上线时间2025-06-10 00:00:00
代码更新2026-01-09
仓库体积4.6 MB
本站数据更新2026-11-11
维护状态低维护
收录时间2026-08-09
浏览次数5

项目介绍

grafting 是 NeurIPS 2025 Oral 论文的官方代码,研究扩散 Transformer(DiT)的设计空间。它提出一种称为“嫁接”的架构编辑方法,将不同 DiT 变体的层或模块进行组合替换,以探索最优设计。项目解决了 DiT 架构设计缺乏系统性探索的问题,通过可复现的实验代码,帮助研究者快速对比不同设计选择(如卷积与注意力混合)对生成质量的影响。核心能力包括架构编辑工具链、实验配置和基准结果,支持在图像生成任务上验证新设计。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 77

核心亮点

  • 官方代码,与 NeurIPS 论文同步,可信度高
  • 提供架构编辑方法,便于系统性探索 DiT 变体
  • 实验配置清晰,易于复现和扩展

不足之处

  • 早期项目,文档和社区支持待观察
  • 依赖 Jupyter Notebook,可能不适合大规模训练

适用场景

  • 研究扩散模型架构设计
  • 对比不同注意力与卷积组合的效果
  • 复现 NeurIPS 论文实验

替代项目

DiT、U-ViT、MDT

上一篇:flow-matching

下一篇:simple-diffusion

同类项目推荐

Fast-LLM 开源

让 LLM 训练跑满 GPU,少等快出结果

Accelerating your LLM training to full speed! Made with ❤️ by ServiceNow Research

★ 339 2026-09-26
mlx-lm-lora 开源

Mac 上微调大模型,LoRA/DPO 一条命令跑通

Train Large Language Models on MLX.

★ 469 2026-09-15
autogluon 开源

三行代码搞定高精度模型,自动调参集成,省时省力

Fast and Accurate ML in 3 Lines of Code

★ 10825 2026-08-10
Awesome-Model-Quantization 开源

一页看全模型量化论文、工具与部署方案

A curated collection of papers, benchmarks, surveys, and tools for model quantizatio···

★ 2465 2026-10-17
CUDA-L2 开源

用强化学习自动写出比 cuBLAS 更快的矩阵乘法内核

CUDA-L2: Surpassing cuBLAS Performance for Matrix Multiplication through Reinforceme···

★ 487 2026-08-09
brewery-ai 开源

零经验也能在控制台微调 AI 模型

An intuitive console agent to fine-tune AI models with little to no experience requi···

★ 213 2026-10-09
AutoGrad-Engine 开源

600 行纯 C# 从零实现 GPT 训练与推理

A complete GPT language model (training and inference) in ~600 lines of pure C#, zer···

★ 403 2026-10-31
OpenPipe 开源

把贵提示词变成便宜微调模型,省下大笔API费用

Turn expensive prompts into cheap fine-tuned models

★ 2839 2026-08-09
denoising_diffusion 开源

用PyTorch Lightning复现DDPM,快速跑通扩散模型训练与生成

Unofficial Implementation of "Denoising Diffusion Probabilistic Models" in PyTorch(L···

★ 91 2026-08-09