Krea 2 完整训练备份
集中保存训练产物、续训状态、数据集、基础权重、缓存、日志及环境记录。
本次训练
- 19 张图片与同名 TXT 标注。
- 分辨率基准:1536 × 1536;开启宽高比分桶,实际桶尺寸随原图比例变化。
- batch size:2;小桶尾批可能只有 1 张。
- 100 epoch,14 batch/epoch,共 1400 optimizer steps。
- Krea 2 RAW,LoRA rank/alpha 32/32,AdamW8bit,学习率 1e-4,BF16。
- 完成日志:1400/1400,最终 avr_loss 0.0663;loss 本身不代表出图质量。
- 包含每轮检查点、最终 LoRA 与最终可续训状态。
最终模型:
runs/krea2_1536_bs2_100ep_20260927_085512/output/krea2_1536_bs2.safetensors
续训状态目录:
runs/krea2_1536_bs2_100ep_20260927_085512/output/krea2_1536_bs2-state/
目录
| 路径 | 内容 |
|---|---|
| models/ | RAW DiT、Qwen-Image VAE、Qwen3-VL 文本编码器 |
| datasets/PNG/ | 19 张训练图片及标注 |
| datasets/PNG.zip | 原始数据集压缩包 |
| runs/krea2_1536_bs2_100ep_20260927_085512/ | 100 轮训练的检查点、最终 LoRA、续训状态、缓存和配置 |
| runs/krea2_20260927/ | 早先 1024 分辨率、16 轮训练的产物、状态、数据和日志 |
| config/ | GUI 和数据集配置快照 |
| logs/ | GUI/训练日志及 TensorBoard 曲线 |
| environment/ | 两套环境的依赖记录、源码版本和源码压缩包、GUI 的 uv.lock |
| runtime/ | 本地启动脚本及本次运行路径记录 |
| manifest.json / SHA256SUMS | 文件大小、SHA-256 校验清单 |
下载与恢复
使用 Hugging Face CLI 下载完整仓库并校验:
hf download ACCC1380/krea2-training-backup --repo-type model --local-dir ./krea2-training-backup
cd krea2-training-backup
sha256sum -c SHA256SUMS
只下载最终 LoRA,可在 Files 页打开上述最终模型路径。
配置与日志保留原环境的绝对路径 /root/musubi-workspace。如恢复到其他目录,需要修改 TOML 中的模型、数据、缓存、输出和日志路径。恢复环境可参考 environment 中的源码版本、依赖列表与 uv.lock;未打包虚拟环境二进制和访问凭据。
训练后端支持 --resume 指向保存的 state 目录,用于恢复优化器等状态。仅加载最终 LoRA 权重与完整恢复训练状态不同。
此备份不包含 Turbo 底模。官方推荐 RAW 训练、Turbo 推理。基础模型及源码沿用各自原有许可;本仓库不重新授予第三方权重或数据的许可。
基础模型来源:
训练工具来源:
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support
Model tree for ACCC1380/krea2-training-backup
Base model
krea/Krea-2-Raw