基于nvidia/cuda镜像,以build模式生成开发环境(预装python3.12、vscode中文语言包、aria2c下载工具等)。之后生成venv并预装torch、flash_attn、sageattention等库。预装部分ComfyUI热门插件,并提供基于Gradio的图形化下载工具。
目前配置为从镜像启动。如需要每次启动重新构建docker环境可以自行修改 .cnb.yml 文件,重新构建的环境不含venv和ComfyUI,可参考venv生成方法和安装ComfyUI及插件。
相关项目链接:ComfyUI 在线体验(飞桨AI Studio星河社区)
新增dev分支,会研究/启用一些实验性功能,可能存在一定的限制!
新增模型自动下载功能,ComfyUI启动后界面内可选择的模型在运行时会自动下载到云节点缓存,无需人为操作,也不需要保存到个人仓库!!!
除了预设模型,还可以在自定义下载链接.json文件里添加自己喜欢的模型,注意json格式!!!
详细代码实现见ComfyUI源码修改,开源项目欢迎复制/参考代码,但希望注明代码出处~
注意:
自动下载功能是修改ComfyUI源码实现的,如果自行更新ComfyUI,可能会失去自动下载功能!!!如需更新ComfyUI可使用内置命令 update 。
自动下载功能涵盖checkpoints、clip、clip_vision、controlnet、diffusion_models、loras、text_encoders、upscale_models、vae文件夹~
对于插件需要的模型暂不支持自动下载,主要是插件有自己的模型查找方式,不一定调用ComfyUI底层提供的模型查找函数。
预置模型来源于AI Models、ComfyUI-yi_dian_tong,不可商用!不可商用!不可商用!
# add 文件或文件夹的路径1 文件或文件夹的路径2 文件或文件夹的路径3 ... # 例如提交comfyui、custom_nodes、venv文件夹 add comfyui custom_nodes venv
注意:可在资源管理器内,右键文件或文件夹,复制路径。
del 文件或文件夹的路径1 文件或文件夹的路径2 文件或文件夹的路径3 ...
qd
可添加其他启动参数,如--use-sage-attention,启用sageattention。
qd --use-sage-attention
注意:--listen、--auto-launch参数已内置,无需额外添加。
ui
update
backup
注意:默认跳过comfyui、custom_nodes、models、venv文件夹,打包格式为.tar.gz。
环境启动后会自动启动ComfyUI。如需关闭,终端中按Ctrl+c。
再次启动ComfyUI,终端输入 qd,回车运行。可添加其他启动参数,如 qd --use-sage-attention,启用sageattention。
四种打开外链的方法(选一种即可):

首先点击右下角 + 打开第二个终端,在新终端中输入 ui,回车运行,将自动打开下载工具界面。如需关闭,对应终端中按Ctrl+c。

Gradio下载工具界面:

预设下载模块中预置了一些模型的下载链接和存放路径,首先选择一级分类和二级分类,然后选择预设文件列表(可多选),最后点击开始下载或生成终端下载代码。
Aria2c和Wget模块均可自定义下载链接和保存路径,对于多数下载链接可以直接识别文件名,无法识别的需要手工输入。
注意:
预设下载中有的模型尽量不要提交到自己的仓库,预设链接是内网链接,下载很快。免费空间只有100+100GB,只要提交文件就会占用空间,后续即使删除文件,空间占用不会降低。需要释放仓库空间只能是删除仓库(注意下载备份重要文件!)。