logo
1
3
Login

云原生构建ai-toolkit运行环境

Fork仓库后点击右上角 启动 L40 预装环境 稍等一会即可开始玩耍~ 点击 执行按钮可探索更多玩法~

多模式启动

1. H20显卡 96GB显存 消耗 32核时/小时

2. L40显卡 48GB显存 消耗 16核时/小时

3. CPU环境 云原生开发-CPU 总额度: 1600核时

pytorch版本,xformers版本,flash_attn版本对照表

torchtorchvisiontorchaudioxformersflash-attnSageAttention
2.7.10.22.12.7.10.0.31.post12.7.4.post12.2.0
2.8.00.23.02.8.00.0.32.post22.8.22.2.0
2.9.00.24.02.9.00.0.33.post12.8.32.2.0
2.9.10.24.12.9.10.0.33.post22.8.32.2.0

vscode环境

1. 集成python插件

2. 集成zsh插件

3. 集成jupyterlab环境

4. 集成nodejs环境

ai-toolkit 环境启动方式

开启云原生开发环境后自动进入环境,退出直接操作快捷键 crtl + c ,默认端口8675做转发访问,通过以下命令手动启动

bash start-ai-toolkit.sh

jupyterlab环境启动方式

开启云原生开发环境后通过终端,执行下面命令启动jupyterlab环境,默认端口8687做转发访问

bash jupyterlab.sh

ComfyUI环境启动方式

开启云原生开发环境后通过终端,执行一下命令启动服务,退出直接操作快捷键 crtl + c ,默认端口8188做转发访问,通过以下命令手动启动

bash start-comfyui.sh

训练模型下载

类型说训练模型类型依赖模型CNB平台收录CNB收录地址
imageChromalodestones/Chroma1-Basehttps://cnb.cool/ai-models/lodestones/Chroma1-Base
Flex.1ostris/Flex.1-alphahttps://cnb.cool/ai-models/ostris/Flex.1-alpha
Flex.2ostris/Flex.2-previewhttps://cnb.cool/ai-models/ostris/Flex.2-preview
Flux.1black-forest-labs/FLUX.1-devhttps://cnb.cool/ai-models/black-forest-labs/FLUX.1-dev
Flux.2black-forest-labs/FLUX.2-devhttps://cnb.cool/ai-models/black-forest-labs/FLUX.2-dev
HiDreamHiDream-ai/HiDream-I1-Fullhttps://cnb.cool/ai-models/HiDream-ai/HiDream-I1-Full
Lumina2Alpha-VLLM/Lumina-Image-2.0https://cnb.cool/ai-models/Alpha-VLLM/Lumina-Image-2.0
OmniGen2OmniGen2/OmniGen2https://cnb.cool/ai-models/OmniGen2/OmniGen2
Qwen-ImageQwen/Qwen-Imagehttps://cnb.cool/ai-models/Qwen/Qwen-Image
SD1.5stable-diffusion-v1-5/stable-diffusion-v1-5https://cnb.cool/ai-models/stable-diffusion-v1-5/stable-diffusion-v1-5
SDXLstabilityai/stable-diffusion-xl-base-1.0https://cnb.cool/ai-models/stabilityai/stable-diffusion-xl-base-1.0
Z-Image-TurboTongyi-MAI/Z-Image-Turbohttps://cnb.cool/ai-models/Tongyi-MAI/Z-Image-Turbo
Z-Image-Turboostris/zimage_turbo_training_adapterhttps://cnb.cool/ai-models/ostris/zimage_turbo_training_adapter
Z-Image-De-Turboostris/Z-Image-De-Turbohttps://cnb.cool/ai-models/ostris/Z-Image-De-Turbo
NSTRUCTIONFLUX.1-Kontext-devblack-forest-labs/FLUX.1-Kontext-devhttps://cnb.cool/ai-models/black-forest-labs/FLUX.1-Kontext-dev
HiDream-E1HiDream-ai/HiDream-E1-1https://cnb.cool/ai-models/HiDream-ai/HiDream-E1-1
Qwen-Image-EditQwen/Qwen-Image-Edithttps://cnb.cool/ai-models/Qwen/Qwen-Image-Edit
Qwen-Image-Edit-2509Qwen/Qwen-Image-Edit-2509https://cnb.cool/ai-models/Qwen/Qwen-Image-Edit-2509
videoWan2.1(1.3B)Wan-AI/Wan2.1-T2V-1.3B-Diffusershttps://cnb.cool/ai-models/Wan-AI/Wan2.1-T2V-1.3B-Diffusers
Wan2.1(14B)Wan-AI/Wan2.1-T2V-14B-Diffusershttps://cnb.cool/ai-models/Wan-AI/Wan2.1-T2V-14B-Diffusers
Wan2.1-I2V(14B-480P)Wan-AI/Wan2.1-I2V-14B-480P-Diffuserhttps://cnb.cool/Wan-AI/Wan2.1-I2V-14B-480P-Diffusers
Wan2.1-I2V(14B-720P)Wan-AI/Wan2.1-I2V-14B-720P-Diffusershttps://cnb.cool/Wan-AI/Wan2.1-I2V-14B-720P-Diffusers
Wan2.2-T2V(14B)ai-toolkit/Wan2.2-T2V-A14B-Diffusers-bf16https://cnb.cool/ai-toolkit/Wan2.2-T2V-A14B-Diffusers-bf16
Wan2.2-I2V(14B)ai-toolkit/Wan2.2-I2V-A14B-Diffusers-bf16https://cnb.cool/ai-toolkit/Wan2.2-I2V-A14B-Diffusers-bf16
Wan2.2-TI2V(5B)Wan-AI/Wan2.2-TI2V-5B-Diffusershttps://cnb.cool/ai-models/Wan-AI/Wan2.2-TI2V-5B-Diffusers

开启云原生开发环境后通过终端,执行命令需要下载类型模型(例如下载z-image模型:下载路径脚本是/workspace/training/image/Z-Image-Turbo-download.sh),模型放在cnb缓存路径/workspace/models_cnb_cache,训练模型时注意填写全路径

bash /workspace/training/image/Z-Image-Turbo-download.sh

特别鸣谢

cnb平台自动下载模型详细代码实现见开源项目作者 ComfyUI源码修改

联系作者

请搜索微信号

About

全系炼丹炉

Language
Shell99.4%
HTML0.6%