mcpField
M
mindspeed-fsdp2-model-migration

mindspeed-fsdp2-model-migration

github.com/ascend开发5273 次安装25 周前更新
代码仓库

关于 mindspeed-fsdp2-model-migration

用于将模型层迁移到 MindSpeed-MM FSDP2 的技能,面向需要新增或修复模型插件的迁移任务。它围绕注册、加载签名、特殊 token 与训练前向契约提供适配要求和检查清单。

主要功能

  1. 01模型插件迁移与实现
  2. 02校验 model_id 注册契约
  3. 03适配模型加载签名
  4. 04保留特殊 token 处理
  5. 05检查训练前向与 loss
  6. 06生成迁移检查清单

适用场景

  1. 01将源模型接入 MindSpeed-MM FSDP2
  2. 02修复模型注册或 ModelHub 兼容性
  3. 03适配 from_pretrained 与 _from_config 调用
  4. 04核对迁移改动并记录适配理由

安装

复制下面的命令,在终端中运行。

$npx skills add https://github.com/ascend/agent-skills --skill mindspeed-fsdp2-model-migration

安装后在 Claude Code 中输入/mindspeed-fsdp2-model-migration确认已加载。

包含的文件

除 SKILL.md 外,安装时一起下载的参考资料和脚本。

checks/
model-acceptance.md
model-contract-gates.md
errors/
model-error-dictionary.md
triage-severity.md
examples/
minimal-io.md

详细介绍

来自项目 README

MindSpeed FSDP2 模型迁移

目标

在满足框架契约的前提下,将模型层迁移到 MindSpeed-MM FSDP2,并保持源模型语义。

适用场景

  • 迁移任务需要在 mindspeed_mm/fsdp/models/ 下创建模型插件。
  • 需要新增或修复 model_id 注册及 ModelHub 兼容性。
  • 需要适配模型加载签名(from_pretrained、_from_config)。

不适用场景

  • 任务是数据预处理或 dataloader 行为调整。
  • 任务仅涉及 YAML 映射。
  • 任务仅为验证阶段问题分诊。

输入

  • 源模型入口文件
  • 目标 model_id
  • 必需的特殊 token 行为
  • 目标配置中的 FSDP2 约束
  • runtime_assets.model_path

输出

  • 模型插件实现
  • 带通过/失败状态的模型迁移检查清单
  • 每一处“非复用改动”的理由说明

必检项

  1. 注册契约:
    • @model_register.register("<model_id>")
  2. 构建契约:
    • from_pretrained 与 ModelArguments 调用链兼容
    • _from_config 路径有效
  3. 语义契约:
    • 源模型需要特殊 token 时,必须保留其处理流程
    • 仅在必要时进行 embedding resize/init
  4. 训练契约:
    • 前向路径可接收 trainer 输入并返回 .loss

防偏差执行协议

  1. 实施前必须完成:
    • 校验 runtime_assets.model_path 存在且可读
    • 对照最相似案例抽取签名模式
  2. 实施后必须产出:
    • model_contract_checklist.md
    • signature_compat_report.md
  3. 失败即停止:
    • from_pretrained 或 _from_config 任一不兼容,禁止继续后续阶段

禁止行为

  • 禁止把 from_pretrained 写成仅单参数签名
  • 禁止仅靠“代码阅读”判定兼容,必须给调用证据
  • 禁止在未确认 token-id 对齐时修改 token 常量
  • 禁止删除源模型关键后处理逻辑而不给出等价替代

实施规则

  • 优先复用源逻辑;仅在框架契约不兼容时做适配。
  • 修改范围仅限模型插件区域。
  • 不引入对源仓包路径的外部硬依赖。
  • 记录每项适配的精确原因。

退出标准

  • 模型可通过 ModelHub 按目标 model_id 构建。
  • 加载签名冲突已解决。
  • 模型侧清单结论均有证据支撑。

渐进式资源

  • checks/model-acceptance.md
  • checks/model-contract-gates.md
  • errors/model-error-dictionary.md
  • errors/triage-severity.md
  • examples/minimal-io.md

仅在验收或错误分诊需要详细判据时读取这些文件。

评论

支持 Markdown