可图_ComfyUI

我要开发同款
匿名用户2024年07月31日
34阅读
开发技术Pytorch
所属分类ai
开源地址https://modelscope.cn/models/cunkai/Kolors_ComfyUI

作品详情

帮助社区的同学快速创建一个可图Kolors_ComfyUI环境

IP适配器工作流

可图默认工作流

点击右上角的Notebook快速开发->使用魔搭平台提供的免费实例->推荐选方式二 GPU环境 -> 启动 -> 查看Notebook

打开后:挨个点击播放按钮,提示安装完成(播放按钮旁边有绿色对号以后),继续点下一个播放按钮

1、安装ComfyUI本体

import os
import subprocess
import sys
import time

# ComfyUI仓库
git_url = "https://github.com/comfyanonymous/ComfyUI.git"

if not os.path.exists('./ComfyUI'):
    subprocess.run(["git", "clone", git_url])
    subprocess.run(["pip", "install", "-r", "./ComfyUI/requirements.txt"])
else:
    print("已经存在ComfyUI")

time.sleep(1)
print('# ComfyUI本体安装完成')

2、安装插件

import os
import subprocess
import sys
import time

def install_requirements(directory):
    # 检查指定目录下是否有requirements.txt文件
    requirements_path = os.path.join(directory, 'requirements.txt')
    if os.path.exists(requirements_path):
        print(f"Found requirements.txt in {directory}. Installing packages...")
        # 使用pip安装requirements.txt中的包
        subprocess.run(['pip', 'install', '-r', requirements_path])
    else:
        print(f"No requirements.txt found in {directory}.")


urls = [
    #manager插件
    "https://gitee.com/cunkai/comfy-ui-manager.git",
    #KwaiKolorsWrapper插件
    "https://github.com/kijai/ComfyUI-KwaiKolorsWrapper.git",
    #ipadapter_plus插件
    "https://github.com/cubiq/ComfyUI_IPAdapter_plus.git",
    #ComfyUI-Kolors-MZ插件
    "https://github.com/MinusZoneAI/ComfyUI-Kolors-MZ.git"
]

for url in urls:
    dir = f"./ComfyUI/custom_nodes/{url.split('/')[-1].replace('.', '_')}"
    subprocess.run(["git", "clone", "--progress", url, dir])
    install_requirements(dir)

time.sleep(1)
print('# 插件安装完成')

3、第三步:下载可图、GLM3、clip、ipadapter  下载时间比较长

import os
import subprocess
import sys
import time

#下载可图unet   5.16GB
!wget -c "https://www.modelscope.cn/models/Kwai-Kolors/Kolors/resolve/master/unet/diffusion_pytorch_model.fp16.safetensors" -O "./ComfyUI/models/unet/kolors-unet.safetensors"

#Vae解码器
!wget -c "https://www.modelscope.cn/models/Kwai-Kolors/Kolors/resolve/master/vae/diffusion_pytorch_model.fp16.bin" -O "./ComfyUI/models/vae/kolors_vae.fp16.bin"

#下载语言模型GLM3  12.5 GB
!mkdir -p ./ComfyUI/models/LLM/
!wget -c "https://www.modelscope.cn/models/cunkai/chatglm3-fp16/resolve/master/chatglm3-fp16.safetensors" -O "./ComfyUI/models/LLM/chatglm3-fp16.safetensors"

#可图需要的clip-vit-large-patch14-336视觉剪辑
!wget -c "https://www.modelscope.cn/models/AI-ModelScope/clip-vit-large-patch14-336/resolve/master/pytorch_model.bin" -O "./ComfyUI/models/clip_vision/clip-vit-large-patch14-336.bin"

#安装ipadapter
!mkdir -p ./ComfyUI/models/ipadapter/
!wget -c "https://www.modelscope.cn/models/cunkai/Kolors_ComfyUI/resolve/master/ip-model/Kolors-IP-Adapter-Plus.bin" -O "./ComfyUI/models/ipadapter/Kolors-IP-Adapter-Plus.bin"

time.sleep(1)
print('# 模型安装完成')

最后一步:启动ComfyUI,出现To see the GUI go to: http://127.0.0.1:8188 点击127.0.0.1:8188打开ComfyUI页面

import subprocess
import torch
import os
import signal


def kill_process8188():
    cmd = "netstat -anp | grep :8188"
    result = os.popen(cmd).read()
    print(result)
    pids = []
    for line in result.splitlines():
        pid = int(line.split()[-1].split('/')[0])
        pids.append(pid)
    for pid in pids:
        os.kill(pid, signal.SIGKILL)

# 结束8188进程
kill_process8188()


if not torch.cuda.is_available():
    print("没有显卡的环境,CPU方式启动")
    subprocess.run(["python", "./ComfyUI/main.py", "--cpu"])
else:
    print("显卡启动")
    subprocess.run(["python", "./ComfyUI/main.py"])
声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论