阿里云ComfyUI使用shell脚本下载Checkpoint模型、Lora模型
柠诺卿
2024年02月24日 00:39
ComfyUI

https://xn--comfyui2:,gpucomfy%20https-6m2ylb6225oucau44a9nzi0mt74a6fdea9184akdas92ic3j7uj3ox8yun13dsn3aub3gz9qx25ctz3hw26ci4rh26c//www.bilibili.com/video/BV1Pc411y7Rr/?share_source=copy_web&vd_source=fd59995bf2f70580369462145819da94

由于该视频教程使用阿里云的是Bucket中的数据集上传进行中转,然后一个模型就有1G以上,从网站下载至电脑本地,然后再上传至数据集过于繁琐。

所以决定编写一个脚本下载模型,支持所有的直链资源下载,可以从LiblibAI提取的直链下载,目前定义以下文件夹目录,方便选择,其他目录可以自己从代码添加

checkpoint

Lora

VAE

controlnet模型

放大模型

《脚本由ChatGPT编写》随便修改搬运

《脚本由ChatGPT编写》随便修改搬运

《脚本由ChatGPT编写》随便修改搬运


「 支持多行直链一并下载」LiblibAI没有的模型,可以通过

https://ty.ly93.cc/

进行转换直链下载


#!/bin/bash

# 定义函数:下载模型并保存

function download_and_save_model() {

  clear

  sleep 0.5

  while true; do

    # 提供选项给用户选择保存地址

    echo "请选择模型保存的地址:"

    echo "1. checkpoint大模型 (ComfyUI/models/checkpoints/)"

    echo "2. Lora (ComfyUI/models/loras/)"

    echo "3. VAE (ComfyUI/models/vae/)"

    echo "4. controlnet模型 (ComfyUI/models/controlnet/)"

    echo "5. 放大模型 (ComfyUI/models/upscale_models/)"

    echo "6. 其他 (模型下载之后将保存在根目录,请自行移动到相关文件夹。)"

    echo "7. 退出程序"

    echo "请输入选项编号:"

    read -r SAVE_OPTION

    case "$SAVE_OPTION" in

      1) SAVE_PATH="ComfyUI/models/checkpoints/" ;;

      2) SAVE_PATH="ComfyUI/models/loras/" ;;

      3) SAVE_PATH="ComfyUI/models/vae/" ;;

      4) SAVE_PATH="ComfyUI/models/controlnet/" ;;

      5) SAVE_PATH="ComfyUI/models/upscale_models/" ;;

      6) echo "模型下载之后将保存在根目录,请自行移动到相关文件夹。" ;;

      7) 

        echo "程序已退出。"

        exit 0

        ;;

      *) echo "无效的选项,请重新输入。" ;;

    esac

    if [[ "$SAVE_OPTION" != "7" ]]; then

      break

    fi

  done

  while true; do

    # 提示用户输入模型的下载地址,并进行检测

    echo "请输入模型的下载地址,每个链接一行,输入完毕后输入 Q 退出当前程序:"

    read -r MODEL_URL

    if [ "$MODEL_URL" == "Q" ] || [ "$MODEL_URL" == "q" ]; then

      echo "程序已退出。"

      exit 0

    elif [[ "$MODEL_URL" != "Q" ]] && [[ "$MODEL_URL" != "q" ]]; then

      # 使用wget下载模型,并设置--content-disposition选项以保存原始文件名

      wget --content-disposition "$MODEL_URL" -P "$SAVE_PATH"

      # 检查下载是否成功

      if [ $? -eq 0 ]; then

        echo "模型下载成功!"

      else

        echo "模型下载失败,请检查输入的下载地址和网络连接。"

      fi

    fi

  done

}

# 调用函数下载模型

download_and_save_model


脚本格式,以及存放地址

启动命令(.sh在根目录的情况下),"d"为执行文件名,可自定义

./d.sh