DiffPipeForge 全模型训练 WebUI

简介

这是面向 Linux 算力实例的 DiffPipeForge 源码训练环境，内置 Python 3.12、PyTorch 2.7.1、CUDA 12.8、完整模型训练模块及常用依赖。实例启动后会自动拉起 WebUI，并启用平台访问认证。

建议使用 24GB 及以上 NVIDIA GPU。classic Cosmos1 不在本镜像支持范围内。

应用入口

• SD-WebUI：端口 7860，用于项目创建、数据集与验证集配置、模型选择和下载、训练、监控及检查点恢复。
• JupyterLab：端口 8888，用于高级文件管理和调试；普通用户无需使用。

首次使用

1. 部署实例并等待状态变为“运行中”。
2. 在实例卡片中打开 SD-WebUI。
3. 新建项目；每个项目使用独立输出目录，检查点互相隔离。
4. 在“数据集设置”中，可从浏览器上传训练数据集，或填写服务器上的数据集文件夹路径。
5. 在“验证集设置”中，先启用验证，再从浏览器上传验证数据集；验证关闭时上传控件不可用。
6. 在“训练配置”中选择模型类型、模型路径与精度，保存配置。
7. 在“开始训练”中检查输出目录、显存参数和检查点恢复路径后启动。

模型与数据

镜像不内置用户模型权重或训练数据。可通过 WebUI 的模型管理功能准备模型，也可将云硬盘中的模型路径填写到配置中。数据集上传后会存放到 WebUI 管理的服务器目录。

模型路径必须指向实际模型文件或模型目录；检查点恢复路径必须指向某次训练输出目录中的检查点文件夹，而不是单个文件。

环境与依赖

• Linux 源码版
• Python 3.12
• PyTorch 2.7.1
• CUDA 12.8
• DeepSpeed、FlashAttention 及各模型所需依赖已安装
• WebUI 开机自动启动
• 构建提交：8d1ce47

显存建议

• 最低建议：24GB NVIDIA GPU
• 大模型或高分辨率训练可能需要更大显存
• 显存不足时优先降低分辨率、批次大小、帧数或启用分块与卸载选项

检查点恢复

“从检查点恢复”使用检查点文件夹。WebUI 会在当前训练会话对应输出目录中探测可用检查点；不同项目与训练会话的输出目录互相隔离，也可以手动填写其他合法检查点文件夹路径。

验证

镜像已完成云端 WebUI 自动启动、HTTP 访问、Python 依赖完整性、训练数据集与验证数据集浏览器上传、验证开关逻辑，以及 JupyterLab 8888 与 SD-WebUI 7860 应用映射检查。

支持范围

支持界面中列出的模型训练模块。classic Cosmos1 明确不包含在本镜像支持范围内。