
Flow-Factory
让流匹配模型轻松学会强化学习,生成更符合目标的结果
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 749
使用教程
ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。
环境要求
- CUDA GPU 24G 以上
- Python 3.10+
安装与启动步骤
-
1克隆
先拉代码。
git clone https://github.com/X-GenGroup/Flow-Factory.git -
2装本包
用可编辑模式装,方便改配置。
pip install -e . -
3装定制扩散库
它依赖改过的扩散库版本。
pip install -e ./diffusers -
4准备数据
按文档把提示词与图片对齐。
# 参考 configs 目录示例 -
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···
sd-lora-trainer
开源
拖入图片,一键训练专属 LoRA,快速定制 SD 模型风格
LoRa trainer for SDXL and SD15
ArcticTraining
开源
配置即跑通 LLM 后训练,省去搭管线的时间
ArcticTraining is a framework designed to simplify and accelerate the post-training ···
transformerlab-app
开源
一站式搞定大模型训练、微调和评估,从笔记本到集群无缝扩展
The open source research environment for AI researchers to seamlessly train, evaluat···
flow-matching
开源
用流匹配快速训练图像生成模型,替代扩散模型
Flow Matching implemented in PyTorch
alpamayo-recipes
开源
自动驾驶大模型微调、量化、部署一条龙配方
Developer Hub for NVIDIA Alpamayo, containing ready-to-use recipes for fine-tuning, ···
awesome-llms-fine-tuning
开源
一站式收藏LLM微调资源,从入门到进阶少走弯路
Explore a comprehensive collection of resources, tutorials, papers, tools, and best ···
LightRFT
开源
用奖励模型驱动微调,让 LLM/VLM 按目标越训越强
LightRFT: Light, Efficient, Omni-modal & Reward-model Driven Reinforcement Fine-Tuni···
modelkit
开源
让 ML 模型开发到上线更省心
Toolkit for developing and maintaining ML models