Flow-Factory

让流匹配模型轻松学会强化学习,生成更符合目标的结果

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 图像生成 🍴 复刻 63 🐛 待处理 55 👀 关注 6
GitHub ↗
GitHub 星标 ★ 749
维护状态 活跃
是否开源 开源
定价模式 🏷️完全免费 开源框架,Apache-2.0 许可证,完全免费,无付费版本。

项目预览

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

项目数据

分类图像生成
开发团队X-GenGroup
定价模式🏷️完全免费
价格说明开源框架,Apache-2.0 许可证,完全免费,无付费版本。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议Apache-2.0
主要语言Python
技术栈/模型diffusers,diffusion,diffusion-model,diffusion-models,fine-tuning,finetuning,flow-matching,flow-matching-models,image-generation,reinforcement-learning,video-generation
GitHub 星标★ 749
上线时间2025-12-18 00:00:00
代码更新2026-11-10
仓库体积24.0 MB
本站数据更新2026-11-10
维护状态活跃
收录时间2026-08-09
浏览次数3

项目介绍

Flow-Factory 是一个面向 Flow-Matching 模型的统一强化学习框架,旨在简化基于流匹配的生成模型(如扩散模型)中的强化学习训练流程。它解决了在图像和视频生成任务中,如何通过强化学习直接优化下游目标(如美学评分、用户偏好)而非仅依赖最大似然估计的问题。核心能力包括:提供统一的 API 封装多种流匹配模型,支持灵活的奖励函数定义,集成常用的 RL 算法(如 PPO),并针对流匹配的采样过程进行优化。该框架降低了在生成模型中应用 RL 的技术门槛,使研究者能快速实验,也便于在图像生成、视频生成等场景中实现更符合人类偏好的生成效果。

Star 增长趋势

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

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

使用教程

ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。

难度:高级 约 60 分钟 部署方式:本地安装 5 步

环境要求

  • CUDA GPU 24G 以上
  • Python 3.10+

安装与启动步骤

  1. 1克隆

    先拉代码。

    git clone https://github.com/X-GenGroup/Flow-Factory.git
  2. 2装本包

    用可编辑模式装,方便改配置。

    pip install -e .
  3. 3装定制扩散库

    它依赖改过的扩散库版本。

    pip install -e ./diffusers
  4. 4准备数据

    按文档把提示词与图片对齐。

    # 参考 configs 目录示例
  5. 5起训练

    先用小配置验证流程。

    # 运行仓库里的训练入口

如何确认成功

训练能跑过一个 epoch,检查点正常保存。

注意事项

  • 它做的是流匹配训练,和常规扩散训练脚本不通用。
  • 定制扩散库必须用它自带的版本。

核心亮点

  • 统一框架,简化了流匹配模型上 RL 的训练流程,上手快
  • 支持图像和视频生成,覆盖场景广
  • 与主流 RL 算法(如 PPO)集成,便于直接优化自定义奖励

不足之处

  • 项目早期,文档和社区支持尚待完善
  • 依赖特定模型结构,迁移到其他流匹配实现可能需要适配

适用场景

  • 图像生成中根据美学评分优化生成质量
  • 视频生成中根据用户偏好调整动态内容
  • 研究流匹配模型的 RL 对齐方法

替代项目

Diffusion-RL、DDPO、DPOK

上一篇:LeapAlign_Code

下一篇:WaDi

同类项目推荐

model-optimization 开源

几行代码压缩模型,让大模型跑进手机和嵌入式设备

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including q···

★ 1583 2026-10-17
sd-lora-trainer 开源

拖入图片,一键训练专属 LoRA,快速定制 SD 模型风格

LoRa trainer for SDXL and SD15

★ 70 2026-08-09
ArcticTraining 开源

配置即跑通 LLM 后训练,省去搭管线的时间

ArcticTraining is a framework designed to simplify and accelerate the post-training ···

★ 293 2026-09-15
transformerlab-app 开源

一站式搞定大模型训练、微调和评估,从笔记本到集群无缝扩展

The open source research environment for AI researchers to seamlessly train, evaluat···

★ 5200 2026-08-10
flow-matching 开源

用流匹配快速训练图像生成模型,替代扩散模型

Flow Matching implemented in PyTorch

★ 131 2026-08-09
alpamayo-recipes 开源

自动驾驶大模型微调、量化、部署一条龙配方

Developer Hub for NVIDIA Alpamayo, containing ready-to-use recipes for fine-tuning, ···

★ 223 2026-09-16
awesome-llms-fine-tuning 开源

一站式收藏LLM微调资源,从入门到进阶少走弯路

Explore a comprehensive collection of resources, tutorials, papers, tools, and best ···

★ 532 2026-08-09
LightRFT 开源

用奖励模型驱动微调,让 LLM/VLM 按目标越训越强

LightRFT: Light, Efficient, Omni-modal & Reward-model Driven Reinforcement Fine-Tuni···

★ 379 2026-09-16
modelkit 开源

让 ML 模型开发到上线更省心

Toolkit for developing and maintaining ML models

★ 149 2026-08-09