TFMQ-DM

给扩散模型瘦身,4-bit量化不降画质,部署提速。

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 图像生成 🍴 复刻 5 👀 关注 6
访问官网 ↗ GitHub ↗
GitHub 星标 ★ 111
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,Apache-2.0许可证,完全免费,可自行部署使用。

项目预览

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

项目数据

分类图像生成
开发团队ModelTC
定价模式🏷️完全免费
价格说明开源项目,Apache-2.0许可证,完全免费,可自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议Apache-2.0
主要语言Jupyter Notebook
技术栈/模型cvpr,cvpr2024,ddim,diffusion-models,highlight,ldm,post-training-quantization,quantization,stable-diffusion,tpami,tpami2025
GitHub 星标★ 111
上线时间2024-03-09 00:00:00
代码更新2025-09-29
仓库体积120.6 MB
本站数据更新2026-11-09
维护状态低维护
收录时间2026-08-09
浏览次数6

项目介绍

TFMQ-DM 是 CVPR 2024 Highlight 和 TPAMI 2025 论文的官方 PyTorch 实现,专注于扩散模型的训练后量化(PTQ)。扩散模型(如 Stable Diffusion)在图像生成上效果惊艳,但参数量大、推理慢,难以部署到边缘设备。TFMQ-DM 通过时间特征维护(Temporal Feature Maintenance)技术,在量化过程中保留关键的时间步特征,显著降低模型位宽(如 4-bit)同时保持生成质量。项目提供完整的量化流程、评估脚本和预训练权重,支持 DDIM、LDM、Stable Diffusion 等主流模型,旨在解决扩散模型在资源受限场景下的高效部署问题,为研究者和工程师提供即用型量化工具。

Star 增长趋势

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

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

核心亮点

  • 提出时间特征维护机制,量化后图像质量损失小,优于现有 PTQ 方法
  • 支持多种扩散模型(DDIM、LDM、Stable Diffusion),通用性强
  • 官方实现,代码完整,含评估脚本和预训练模型,易于复现

不足之处

  • 项目星标仅110,社区活跃度低,文档和示例有待完善
  • 量化主要针对 UNet 等组件,对完整 pipeline 的端到端加速效果未充分展示

适用场景

  • 边缘设备部署 Stable Diffusion,降低内存和延迟
  • 在保持生成质量的前提下压缩扩散模型,用于移动端图像编辑
  • 研究扩散模型量化技术,作为 PTQ 基线对比

替代项目

Q-Diffusion、PTQ4DM、QuantiTune

同类项目推荐

multimodal 开源

用 PyTorch 快速搭多模态模型训练管线

TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-···

★ 1744 2026-10-17
Soup 开源

一个YAML搞定大模型微调,4GB显卡也能训8B模型

Fine-tune LLMs from one YAML. Layer streaming trains an 8B model on a 4 GB laptop GP···

★ 10332 2026-08-09
xmtf 开源

多任务微调,让模型学会跨语言举一反三

Crosslingual Generalization through Multitask Finetuning

★ 534 2026-08-09
mlx-tune 开源

在 Mac 上本地微调大模型,多模态全支持,速度飞快

Fine-tune LLMs on your Mac with Apple Silicon. SFT, DPO, GRPO, Vision, TTS, STT, Emb···

★ 1456 2026-08-09
MARS 开源

用方差缩减技术,让大模型训练更快更稳

The official implementation of MARS: Unleashing the Power of Variance Reduction for ···

★ 725 2026-08-09
Adam-mini 开源

用更少学习率,训更快更大模型,省显存提性能。

Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406···

★ 462 2026-08-09
POInfra 开源

让扩散/流模型像策略一样在线微调与规划

Policy Optimization for Generative Models: diffusion and flow policies, online fine-···

★ 100 2026-10-14
ai8x-training 开源

在电脑上训练量化模型,一键部署到毫瓦级 AI 芯片

Model Training for ADI's MAX78000 and MAX78002 Edge AI Devices

★ 117 2026-09-19
pycaret 开源

几行代码跑完建模全流程,小白也能交出好模型

Open-source, low-code AutoML platform for Python. PyCaret 4.0: sklearn-native engine···

★ 9859 2026-08-09