文件最后提交记录最后更新时间
[Feature] Add t2v and t2av fine-tuning for LTX 2.0 with FSDP2 backend support Co-authored-by: hanyyy<hanyue42@huawei.com> # message auto-generated for no-merge-commit merge: !2301 merge ltx2_0324 into master [Feature] Add t2v and t2av fine-tuning for LTX 2.0 with FSDP2 backend support Created-by: vasileone Commit-by: hanyyy Merged-by: ascend-robot Description: ## What this PR does / why we need it? 本 PR 在 MindSpeed-MM 的 FSDP2 后端新增 LTX2.0 微调训练能力,提供可直接跑通的 t2v 与 t2av 示例,并补齐视频 / 音视频两套训练 pipeline 模型侧: - 新增 model_id: ltx2,适配 FSDP2 后端 - 新增视频 pipeline - 新增音视频 pipeline ## Modification - mindspeed_mm/fsdp/models/ltx2/ltx2_fsdp2/modeling_ltx2.py:LTX2 的 FSDP2 训练封装 LTX2ForTraining。 - mindspeed_mm/fsdp/data/datasets/ltx2/ltx2_precomputed_dataset.py:ltx2_precomputed fsdp2后端数据集实现。 - examples/fsdp2/ltx2/finetune_ltx2_t2v.sh:t2v torchrun 启动脚本。 - examples/fsdp2/ltx2/finetune_ltx2_t2av.sh:t2av torchrun 启动脚本。 See merge request: Ascend/MindSpeed-MM!23012 个月前