mirror of
https://github.com/kodxana/madiator-docker-runpod.git
synced 2024-11-21 18:30:12 +01:00
Added flux installer for comfy ui
This commit is contained in:
parent
d6186ec00a
commit
bed81abfef
3 changed files with 70 additions and 2 deletions
|
@ -91,5 +91,9 @@ RUN chmod +x /start.sh
|
||||||
RUN chmod +x /pre_start.sh
|
RUN chmod +x /pre_start.sh
|
||||||
COPY comfyui_extras.ipynb /comfyui_extras.ipynb
|
COPY comfyui_extras.ipynb /comfyui_extras.ipynb
|
||||||
|
|
||||||
|
# Add the new install-flux.sh script
|
||||||
|
COPY install-flux.sh /install-flux.sh
|
||||||
|
RUN chmod +x /install-flux.sh
|
||||||
|
|
||||||
# CMD
|
# CMD
|
||||||
CMD [ "/start.sh" ]
|
CMD [ "/start.sh" ]
|
||||||
|
|
63
official-templates/better-comfyui/install-flux.sh
Normal file
63
official-templates/better-comfyui/install-flux.sh
Normal file
|
@ -0,0 +1,63 @@
|
||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
# Function to download file
|
||||||
|
download_file() {
|
||||||
|
local url="$1"
|
||||||
|
local destination="$2"
|
||||||
|
local token="$3"
|
||||||
|
|
||||||
|
if [ -f "$destination" ]; then
|
||||||
|
read -p "File $destination already exists. Overwrite? (y/n): " overwrite
|
||||||
|
if [[ $overwrite != [Yy]* ]]; then
|
||||||
|
echo "Skipping download of $destination"
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -n "$token" ]; then
|
||||||
|
wget --header="Authorization: Bearer $token" -O "$destination" "$url"
|
||||||
|
else
|
||||||
|
wget -O "$destination" "$url"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# Check for Hugging Face token
|
||||||
|
if [ -z "$HF_TOKEN" ]; then
|
||||||
|
read -p "Enter your Hugging Face token (press Enter to skip for public models): " HF_TOKEN
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Menu for model selection
|
||||||
|
echo "Select Flux AI Model variant to install:"
|
||||||
|
echo "1. Single-file FP8 version (16GB VRAM)"
|
||||||
|
echo "2. Flux Schnell version (16GB VRAM)"
|
||||||
|
echo "3. Regular FP16 full version (24GB VRAM)"
|
||||||
|
read -p "Enter your choice (1-3): " model_choice
|
||||||
|
|
||||||
|
case $model_choice in
|
||||||
|
1)
|
||||||
|
download_file "https://huggingface.co/Comfy-Org/flux1-dev/resolve/main/flux1-dev-fp8.safetensors" "/workspace/ComfyUI/models/unet/flux1-dev-fp8.safetensors" "$HF_TOKEN"
|
||||||
|
;;
|
||||||
|
2)
|
||||||
|
download_file "https://huggingface.co/Comfy-Org/flux1-schnell/resolve/main/flux1-schnell-fp8.safetensors" "/workspace/ComfyUI/models/unet/flux1-schnell-fp8.safetensors" "$HF_TOKEN"
|
||||||
|
;;
|
||||||
|
3)
|
||||||
|
if [ -z "$HF_TOKEN" ]; then
|
||||||
|
echo "Hugging Face token is required for this model."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
download_file "https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/flux1-dev.safetensors" "/workspace/ComfyUI/models/unet/flux1-dev.safetensors" "$HF_TOKEN"
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
echo "Invalid choice. Exiting."
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
# Download CLIP models
|
||||||
|
download_file "https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/clip_l.safetensors" "/workspace/ComfyUI/models/clip/clip_l.safetensors" "$HF_TOKEN"
|
||||||
|
download_file "https://huggingface.co/comfyanonymous/flux_text_encoders/resolve/main/t5xxl_fp8_e4m3fn.safetensors" "/workspace/ComfyUI/models/clip/t5xxl_fp8_e4m3fn.safetensors" "$HF_TOKEN"
|
||||||
|
|
||||||
|
# Download VAE model
|
||||||
|
download_file "https://huggingface.co/black-forest-labs/FLUX.1-schnell/resolve/main/ae.safetensors" "/workspace/ComfyUI/models/vae/ae.safetensors" "$HF_TOKEN"
|
||||||
|
|
||||||
|
echo "Installation complete!"
|
|
@ -30,9 +30,10 @@ done
|
||||||
|
|
||||||
LOG_FILE="/workspace/comfyui.log"
|
LOG_FILE="/workspace/comfyui.log"
|
||||||
|
|
||||||
# Copy the notebook to the /workspace directory
|
# Copy the notebook and install-flux.sh script to the /workspace directory
|
||||||
print_feedback "Copying notebook to /workspace..."
|
print_feedback "Copying notebook and install script to /workspace..."
|
||||||
cp /comfyui_extras.ipynb /workspace/
|
cp /comfyui_extras.ipynb /workspace/
|
||||||
|
cp /install-flux.sh /workspace/
|
||||||
|
|
||||||
# Check if the NO_SYNC variable is set to true
|
# Check if the NO_SYNC variable is set to true
|
||||||
if [ "${NO_SYNC}" == "true" ]; then
|
if [ "${NO_SYNC}" == "true" ]; then
|
||||||
|
|
Loading…
Reference in a new issue