
https://xn--comfyui2:,gpucomfy%20https-6m2ylb6225oucau44a9nzi0mt74a6fdea9184akdas92ic3j7uj3ox8yun13dsn3aub3gz9qx25ctz3hw26ci4rh26c//www.bilibili.com/video/BV1Pc411y7Rr/?share_source=copy_web&vd_source=fd59995bf2f70580369462145819da94
由于该视频教程使用阿里云的是Bucket中的数据集上传进行中转,然后一个模型就有1G以上,从网站下载至电脑本地,然后再上传至数据集过于繁琐。
所以决定编写一个脚本下载模型,支持所有的直链资源下载,可以从LiblibAI提取的直链下载,目前定义以下文件夹目录,方便选择,其他目录可以自己从代码添加
《脚本由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
