mirror of
https://github.com/community-scripts/ProxmoxVE.git
synced 2026-10-24 04:05:03 +00:00
MickLesk
3f0dd1b09b
Every interactive prompt takes a var_<app>_<name> override and treats a closed stdin like Enter (`|| true`), so an install keeps going after the console that started it is gone. Values the script cannot do without end with exit 64 and name the variable to set. Validation loops stop asking when nobody can answer instead of spinning on an empty read.
104 lines
3.1 KiB
Bash
104 lines
3.1 KiB
Bash
#!/usr/bin/env bash
|
|
|
|
# Copyright (c) 2021-2026 community-scripts ORG
|
|
# Author: jdacode
|
|
# License: MIT | https://github.com/community-scripts/ProxmoxVE/raw/main/LICENSE
|
|
# Source: https://github.com/comfyanonymous/ComfyUI
|
|
|
|
source /dev/stdin <<<"$FUNCTIONS_FILE_PATH"
|
|
color
|
|
verb_ip6
|
|
catch_errors
|
|
setting_up_container
|
|
network_check
|
|
update_os
|
|
setup_hwaccel
|
|
|
|
echo
|
|
echo "${TAB3}Choose the GPU type for ComfyUI:"
|
|
echo "${TAB3}[1]-None [2]-NVIDIA [3]-AMD [4]-Intel"
|
|
gpu_choice="${var_comfyui_gpu_type:-}"
|
|
[[ -n "$gpu_choice" ]] || read -rp "${TAB3}Enter your choice [1-4] (default: 1): " gpu_choice || true
|
|
gpu_choice=${gpu_choice:-1}
|
|
case "$gpu_choice" in
|
|
1) comfyui_gpu_type="none" ;;
|
|
2) comfyui_gpu_type="nvidia" ;;
|
|
3) comfyui_gpu_type="amd" ;;
|
|
4) comfyui_gpu_type="intel" ;;
|
|
*)
|
|
comfyui_gpu_type="none"
|
|
echo "${TAB3}Invalid choice. Defaulting to ${comfyui_gpu_type}."
|
|
;;
|
|
esac
|
|
echo
|
|
|
|
PYTHON_VERSION="3.12" setup_uv
|
|
|
|
fetch_and_deploy_gh_release "ComfyUI" "comfyanonymous/ComfyUI" "tarball" "latest" "/opt/ComfyUI"
|
|
|
|
msg_info "Python dependencies"
|
|
$STD uv venv --clear "/opt/ComfyUI/venv"
|
|
|
|
if [[ "${comfyui_gpu_type,,}" == "nvidia" ]]; then
|
|
pytorch_url="https://download.pytorch.org/whl/cu130"
|
|
if [[ -f "/opt/ComfyUI/README.md" ]]; then
|
|
extracted=$(grep -oP 'pip install.*?--extra-index-url\s+\Khttps://download\.pytorch\.org/whl/cu\d+' /opt/ComfyUI/README.md | head -1 || true)
|
|
[[ -n "$extracted" ]] && pytorch_url="$extracted"
|
|
fi
|
|
$STD uv pip install \
|
|
torch \
|
|
torchvision \
|
|
torchaudio \
|
|
--extra-index-url "$pytorch_url" \
|
|
--python="/opt/ComfyUI/venv/bin/python"
|
|
elif [[ "${comfyui_gpu_type,,}" == "amd" ]]; then
|
|
pytorch_url="https://download.pytorch.org/whl/rocm6.4"
|
|
if [[ -f "/opt/ComfyUI/README.md" ]]; then
|
|
extracted=$(grep -oP 'pip install.*?--index-url\s+\Khttps://download\.pytorch\.org/whl/rocm[\d.]+' /opt/ComfyUI/README.md | grep -v 'nightly' | head -1 || true)
|
|
[[ -n "$extracted" ]] && pytorch_url="$extracted"
|
|
fi
|
|
$STD uv pip install \
|
|
torch \
|
|
torchvision \
|
|
torchaudio \
|
|
--index-url "$pytorch_url" \
|
|
--python="/opt/ComfyUI/venv/bin/python"
|
|
elif [[ "${comfyui_gpu_type,,}" == "intel" ]]; then
|
|
pytorch_url="https://download.pytorch.org/whl/xpu"
|
|
if [[ -f "/opt/ComfyUI/README.md" ]]; then
|
|
extracted=$(grep -oP 'pip install.*?--index-url\s+\Khttps://download\.pytorch\.org/whl/xpu' /opt/ComfyUI/README.md | head -1 || true)
|
|
[[ -n "$extracted" ]] && pytorch_url="$extracted"
|
|
fi
|
|
$STD uv pip install \
|
|
torch \
|
|
torchvision \
|
|
torchaudio \
|
|
--index-url "$pytorch_url" \
|
|
--python="/opt/ComfyUI/venv/bin/python"
|
|
fi
|
|
$STD uv pip install -r "/opt/ComfyUI/requirements.txt" --python="/opt/ComfyUI/venv/bin/python"
|
|
msg_ok "Python dependencies"
|
|
|
|
msg_info "Creating Service"
|
|
cat <<EOF >/etc/systemd/system/comfyui.service
|
|
[Unit]
|
|
Description=ComfyUI Service
|
|
After=network.target
|
|
|
|
[Service]
|
|
Type=simple
|
|
User=root
|
|
WorkingDirectory=/opt/ComfyUI
|
|
ExecStart=/opt/ComfyUI/venv/bin/python /opt/ComfyUI/main.py --listen --port 8188
|
|
Restart=on-failure
|
|
|
|
[Install]
|
|
WantedBy=multi-user.target
|
|
EOF
|
|
systemctl enable -q --now comfyui
|
|
msg_ok "Created Service"
|
|
|
|
motd_ssh
|
|
customize
|
|
cleanup_lxc
|