镜像名称

Krea2 AI-tookit Lora 训练镜像

镜像简介

该镜像基于cuda12.9+pytorch2.8，安装aitookit，内置krea2-raw模型与训练数据集。支持用户对图片生成效果进行个性化微调，实现风格迁移与角色一致性训练

环境与依赖

本镜像构建和运行所需的基础环境。

• 框架及版本：PyTorch 2.8
• CUDA版本：CUDA 12.9
• 其他依赖：Python 3.12

使用说明

系统设置自启动，启动后，通过以下链接即可访问 Web UI， http://公网IP:8675/

如果发现镜像启动之后5分钟，仍然无法访问 AI Toolkit。可以尝试重启镜像。 或者手动启动aiktookit，启动方法如下

实例创建后，打开jupyterlab，按照顺序依次执行以下指令，进入/workspace/ai-toolkit/ui目录

conda activate py312 #系统默认激活，一般情况下不用手动操作

cd /workspace #进入workspace目录，系统默认所在目录

cd ai-toolkit/ui

将“super_secure_password”部分替换为想要设置的密码

AI_TOOLKIT_AUTH=super_secure_password npm run build_and_start  #设置访问密码并启动aitookit

相关链接

• 项目源码： AI tookit GitHub Repository