已合并
feat: support Flux GRPO example #138
feat: support Flux GRPO example #138
已合并
luo-tong-an创建于 6月8日
luo-tong-an
luo-tong-an
6月8日

概述

本 PR 为支持 Flux 模型 GRPO 后训练,在多模态 RL 中新增样例。

类型

  • 新功能
  • 文档内容更新

主要改动

  • 在 diffusers 中修改使用昇腾亲和排布和昇腾亲和算子
  • 在 DanceGRPO 中修改适配昇腾 NPU
  • 在 DanceGRPO 中新增 rollout mbs > 1 功能
  • 在 DanceGRPO 中新增 train mbs > 1 功能

验证

  • 已对 patch 涉及的 Python 文件进行静态 AST 语法检查,检查通过。
likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 luo-tong-an 的贡献)
luo-tong-anluo-tong-an
6月8日 创建了 pull request,commit d9b6b50c
CANN-robotCANN-robot成员
6月8日 添加了label:cann-cla/no
CANN-robot
CANN-robot成员
6月8日 评论:

Thanks for your pull-request.
The full list of commands accepted by me can be found at here
You can get sig-info at here


PR Approval Progress

Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

module lgtm status approve status
repo-cann/cann-recipes-train 张凯宇, zhanghz1 (2/2) 张凯宇 (1/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

luo-tong-an, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
luo-tong-anluo-tong-an
6月8日 修改了pull request 的描述
kkodoo
kkodoo
6月11日 评论:

compile

likedislike
此处折叠了72条消息 查看更多
zhangky
zhangky成员
21 天前 评论:

/approve

likedislike
CANN-robotCANN-robot成员
21 天前 添加了label:approved
zhanghz1
zhanghz1成员
21 天前 评论:

/lgtm

likedislike
CANN-robotCANN-robot成员
21 天前 添加了label:lgtm
CANN-robotCANN-robot成员
21 天前 合入了pull request