fiojanea
/

Esan_testi

+#!/bin/bash
+#!/bin/bash
+source /venv/main/bin/activate
+COMFYUI_DIR=${WORKSPACE}/ComfyUI
+# Packages are installed after nodes so we can fix them...
+APT_PACKAGES=(
+    #"package-1"
+    #"package-2"
+)
+PIP_PACKAGES=(
+    #"package-1"
+    #"package-2"
+)
+NODES=(
+    "https://github.com/ltdrdata/ComfyUI-Manager"
+    "https://github.com/cubiq/ComfyUI_essentials"
+    "https://github.com/ShmuelRonen/ComfyUI-WanVideoKsampler"
+    "https://github.com/kijai/ComfyUI-WanVideoWrapper"
+    "https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite"
+    "https://github.com/DoctorDiffusion/ComfyUI-MediaMixer"
+    "https://github.com/rgthree/rgthree-comfy"
+    "https://github.com/GTSuya-Studio/ComfyUI-Gtsuya-Nodes"
+    "https://github.com/JaredTherriault/ComfyUI-JNodes"
+    "https://github.com/heshengtao/comfyui_LLM_party"
+    "https://github.com/kijai/ComfyUI-KJNodes"
+    "https://github.com/TinyTerra/ComfyUI_tinyterraNodes"
+    "https://github.com/omar92/ComfyUI-QualityOfLifeSuit_Omar92"
+    "https://github.com/bash-j/mikey_nodes"
+    "https://github.com/crystian/comfyui-crystools"
+    "https://github.com/yolain/ComfyUI-Easy-Use"
+    "https://github.com/Fannovel16/ComfyUI-Frame-Interpolation"
+    "https://github.com/LAOGOU-666/Comfyui-Memory_Cleanup"
+    "https://github.com/evanspearman/ComfyMath"
+)
+WORKFLOWS=(
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.1_i2v_720P.json"
+)
+CHECKPOINT_MODELS=(
+    #"https://civitai.com/api/download/models/798204?type=Model&format=SafeTensor&size=full&fp=fp16"
+    #"https://huggingface.co/Kijai/HunyuanVideo_comfy/resolve/main/hunyuan_video_FastVideo_720_fp8_e4m3fn.safetensors"
+    #
+    #"https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan_2.1_vae.safetensors"
+    #"https://huggingface.co/fiojanea/Esan_testi/resolve/main/umt5_xxl_fp16.safetensors"
+    #"https://huggingface.co/fiojanea/Esan_testi/resolve/main/Phantom-Wan-14B_fp16.safetensors"
+    #"https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.1_i2v_720p_14B_bf16.safetensors"
+    #"https://huggingface.co/fiojanea/Esan_testi/resolve/main/clip_vision_h.safetensors"
+)
+UNET_MODELS=(
+)
+LORA_MODELS=(
+  # WAN accelerators ------------------------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/high_noise_model.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/low_noise_model.safetensors"
+ # WAN general ------------------------------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/NSFW-22-H-e8.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/NSFW-22-L-e8.safetensors"
+  # WAN Body modifications ------------------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/genitals_helper_v1.0_e219.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/small_breasts_v1.0_wan14b_epoch050.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Wan_Breast_Helper_Hearmeman.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Wan_Pussy_LoRA_Hearmeman.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/TorpedoTits_wan22_e20_low.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Flat_chest_V1..safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/SaggyTits_wan22_e25_low.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/BBFWv1.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Wan2.2-T2V-Femaled-Vaginus-High-v1.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Wan2.2-T2V-Femaled-Vaginus-Low-V1.safetensors"
+  #"https://huggingface.co/fiojanea/Esan_testi/resolve/main/T2V_D-Cups-14B.safetensors"
+  # WAN Kissing -----------------------------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Wan2.2-T2V-Kissing-HIGH-14B.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Wan2.2-T2V-Kissing-LOW-14B.safetensors"
+  # WAN Undressing --------------------------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/woman_undressing_v1.0_wan14b_epoch048.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/str1p_v1.safetensors"
+  # WAN Penis -------------------------------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/erect_penis_epoch_80.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/dicks_epoch_100.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.1-i2v-480p-penis-arousal-v1.0.safetensors"
+  # WAN Cumshot -----------------------------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.2_highnoise_cumshot_v.1.0.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.2_lownoise_cumshot_v1.0.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan-thiccum-v3.safetensors"
+  # WAN Insertion ----------------------------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.2-i2v-high-oral-insertion-v1.0.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.2-i2v-low-oral-insertion-v1.0.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.2-i2v-high-pov-insertion-v1.0.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.2-i2v-low-pov-insertion-v1.0.safetensors"
+  #"https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.1-i2v-480p-anal-insertion-v1.0.safetensors"
+  # WAN Blowjob ------------------------------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/jfj-deepthroat-W22-I2V-HN.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/jfj-deepthroat-W22-I2V-LN.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Wan2.2-I2V-Blowjob-HIGH-14B.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Wan2.2-I2V-Blowjob-LOW-14B.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/DR34MJOB_I2V_14b_HighNoise.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/DR34MJOB_I2V_14b_LowNoise.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2_2_tongue_job_exp_high_noise.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2_2_tongue_job_exp_low_noise.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/FF-v3-high-120.safetensors"
+  # WAN Masturbating, fingering, posing --------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/WAN-2.2-I2V-BreastPlay-HIGH-v2.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/WAN-2.2-I2V-BreastPlay-LOW-v2.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/W22_NSFW_Posing_Nude_i2v_HN_v1.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/W22_NSFW_Posing_Nude_i2v_LN_v1.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/I2V_-_Dildo_Masturbation_-_14B.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Sensual_fingering_v1_high_noise.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Sensual_fingering_v1_low_noise.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Beingfingered.safetensors"
+  # WAN Fucking -------------------------------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan_pov_missionary_i2v_v1.1.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/analMissionary_1.1_i2v_e100.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.2_i2v_highnoise_pov_missionary_v1.0.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.2_i2v_lownoise_pov_missionary_v1.0.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Wan2.2-T2V-DoggyStyle-HIGH-14B.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Wan2.2-T2V-DoggyStyle-LOW-14B.safetensors"
+  # WAN Spanking, slapping --------------------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/spanking_for_wan_v1_e128.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Whippingfurry.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/slap.safetensors"
+  # WAN Orgasm --------------------------------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Wan2.2-I2V-Orgasm-HIGH-14B.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Wan2.2-I2V-Orgasm-LOW-14B.safetensors"
+  # WAN Spreading -----------------------------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/pussy_spread_i2v_480p.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/ass_spread_i2v_480p_v02.safetensors"
+  # WAN Handjob -------------------------------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Handjob-wan-e38.safetensors"
+  # WAN Lesbian -------------------------------------------------------------------------------------------
+  #"https://huggingface.co/fiojanea/Esan_testi/resolve/main/T2V_lesbian-cunnilingus_v2-14b.safetensors"
+  # WAN Misc -----------------------------------------------------------------------------------------------
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/crying-high.safetensors"
+  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Hit_the_face.safetensors"
+  # WAN Misc (NOT IN USE) ----------------------------------------------------------------------------------
+  # "https://huggingface.co/fiojanea/Esan_testi/resolve/main/WAN-facefuck-epoch110_fast_speed_full_insertion_man_has_sex_with_womans_mouth.safetensors"
+  # "https://huggingface.co/fiojanea/Esan_testi/resolve/main/W_NSFW_Posing_Nude_t2v_14b_v1.safetensors"
+  # "https://huggingface.co/fiojanea/Esan_testi/resolve/main/360_epoch20.safetensors"
+  # "https://huggingface.co/fiojanea/Esan_testi/resolve/main/P007-Catwalk-i2v-v10-000008_converted.safetensors"
+  # "https://huggingface.co/fiojanea/Esan_testi/resolve/main/photoshooting.safetensors"
+  # "https://huggingface.co/fiojanea/Esan_testi/resolve/main/DR34M1SH_I2V_V1.420.safetensors"
+  # "https://huggingface.co/fiojanea/Esan_testi/resolve/main/layuan_zoom-out.safetensors"
+  # "https://huggingface.co/fiojanea/Esan_testi/resolve/main/front_doggy_plow_v1_1_wan.safetensors"
+  # "https://huggingface.co/fiojanea/Esan_testi/resolve/main/I2V-Doggystyle_From_Front-720p.safetensors"
+  # WAN charachters ------------------------------------------------------------------------------------------------
+   "https://huggingface.co/fiojanea/Esan_testi/resolve/main/dak0ta.safetensors"
+ #  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/r11kka.safetensors"
+   "https://huggingface.co/fiojanea/Esan_testi/resolve/main/r1tva.safetensors"
+   "https://huggingface.co/fiojanea/Esan_testi/resolve/main/ann11na_ai.safetensors"
+ #  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/m3rja.safetensors"
+   "https://huggingface.co/fiojanea/Esan_testi/resolve/main/m5w.safetensors"
+   "https://huggingface.co/fiojanea/Esan_testi/resolve/main/p1nja.safetensors"
+   "https://huggingface.co/fiojanea/Esan_testi/resolve/main/r11kka2.safetensors"
+   "https://huggingface.co/fiojanea/Esan_testi/resolve/main/m3rja2.safetensors"
+  # FLUX -----------------------------------------------------------------------------------------------------------
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Flux_NSFW_lora.safetensors"
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/JD3s_Nudify_Kontext.safetensors"
+ #  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/skin_texture_style_v5.safetensors"
+ #  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/skin_tone_style_v2-step00001500.safetensors"
+ #  "https://huggingface.co/fiojanea/Esan_testi/resolve/main/nsfw_flux_lora_v1.safetensors"
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/undressing_flux_v3.safetensors"
+ #   "https://huggingface.co/fiojanea/Esan_testi/resolve/main/OnOffFlux4-9.safetensors"
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/breast-size2.safetensors"
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Flux-Bumpy_nipples.safetensors"
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Flux-Conical_nipples.safetensors"
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Flux-Perky_nipples.safetensors"
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Flux-Small_Nipples.safetensors"
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Flux-Wide_nipples.safetensors"
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Flux-puffy_nipples.safetensors"
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/FLUX_FD-BreastMegaPack-R16.safetensors"
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Flux-Realistic_Pussy.safetensors"
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/Flux-realistic_HDR_v1.safetensors"
+)
+VAE_MODELS=(
+  )
+ESRGAN_MODELS=(
+)
+CONTROLNET_MODELS=(
+)
+### DO NOT EDIT BELOW HERE UNLESS YOU KNOW WHAT YOU ARE DOING ###
+function provisioning_start() {
+  provisioning_print_header
+  provisioning_get_apt_packages
+  provisioning_get_nodes
+  provisioning_get_pip_packages
+  # Download to specific directories
+  provisioning_get_files "${COMFYUI_DIR}/models/vae" \
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan_2.1_vae.safetensors"
+  provisioning_get_files "${COMFYUI_DIR}/models/vae" \
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/ae.safetensors"
+  provisioning_get_files "${COMFYUI_DIR}/models/clip" \
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/clip_l.safetensors"
+  provisioning_get_files "${COMFYUI_DIR}/models/clip" \
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/t5xxl_fp16.safetensors"
+  provisioning_get_files "${COMFYUI_DIR}/models/clip_vision" \
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/clip_vision_h.safetensors"
+  provisioning_get_files "${COMFYUI_DIR}/models/clip_vision" \
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/open_clip_model.safetensors"
+  provisioning_get_files "${COMFYUI_DIR}/models/upscale_models" \
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/RealESRGAN_x4plus.pth"
+  provisioning_get_files "${COMFYUI_DIR}/models/upscale_models" \
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/RealESRGAN_x2plus.pth"
+  provisioning_get_files "${COMFYUI_DIR}/models/diffusion_models" \
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.2_i2v_high_noise_14B_fp8_scaled.safetensors"
+  provisioning_get_files "${COMFYUI_DIR}/models/diffusion_models" \
+    # "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.2_t2v_high_noise_14B_fp8_scaled.safetensors"
+  provisioning_get_files "${COMFYUI_DIR}/models/diffusion_models" \
+    # "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.2_t2v_low_noise_14B_fp8_scaled.safetensors"
+  provisioning_get_files "${COMFYUI_DIR}/models/diffusion_models" \
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/wan2.2_i2v_low_noise_14B_fp8_scaled.safetensors"
+  provisioning_get_files "${COMFYUI_DIR}/models/diffusion_models" \
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/flux1-kontext-dev.safetensors"
+  provisioning_get_files "${COMFYUI_DIR}/models/diffusion_models" \
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/flux1-fill-dev.safetensors"
+  provisioning_get_files "${COMFYUI_DIR}/models/text_encoders" \
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/umt5_xxl_fp8_e4m3fn_scaled.safetensors"
+    #"https://huggingface.co/fiojanea/Esan_testi/resolve/main/umt5_xxl_fp16.safetensors"
+   provisioning_get_files "${COMFYUI_DIR}/models/text_encoders" \
+    "https://huggingface.co/fiojanea/Esan_testi/resolve/main/umt5_xxl_fp16.safetensors"
+  # Other model categories
+  provisioning_get_files "${COMFYUI_DIR}/models/checkpoints" "${CHECKPOINT_MODELS[@]}"
+  provisioning_get_files "${COMFYUI_DIR}/models/loras" "${LORA_MODELS[@]}"
+  provisioning_print_end
+}
+function provisioning_get_apt_packages() {
+    if [[ -n $APT_PACKAGES ]]; then
+            sudo $APT_INSTALL ${APT_PACKAGES[@]}
+    fi
+}
+function provisioning_get_pip_packages() {
+    if [[ -n $PIP_PACKAGES ]]; then
+            pip install --no-cache-dir ${PIP_PACKAGES[@]}
+    fi
+}
+function provisioning_get_nodes() {
+    for repo in "${NODES[@]}"; do
+        dir="${repo##*/}"
+        path="${COMFYUI_DIR}/custom_nodes/${dir}"
+        requirements="${path}/requirements.txt"
+        printf "Cloning or updating node: %s\n" "${repo}"
+        rm -rf "${path}"  # Ensure fresh clone
+        git clone "${repo}" "${path}" --recursive
+        if [[ -e "$requirements" ]]; then
+            echo "Installing requirements for ${dir}..."
+            pip install --no-cache-dir -r "$requirements"
+        fi
+    done
+}
+function provisioning_get_files() {
+    if [[ -z $2 ]]; then return 1; fi
+    dir="$1"
+    mkdir -p "$dir"
+    shift
+    arr=("$@")
+    printf "Downloading %s model(s) to %s...\n" "${#arr[@]}" "$dir"
+    for url in "${arr[@]}"; do
+        printf "Downloading: %s\n" "${url}"
+        provisioning_download "${url}" "${dir}"
+        printf "\n"
+    done
+}
+function provisioning_print_header() {
+    printf "\n##############################################\n#                                            #\n#          Provisioning container            #\n#                                            #\n#         This will take some time           #\n#                                            #\n# Your container will be ready on completion #\n#                                            #\n##############################################\n\n"
+}
+function provisioning_print_end() {
+    printf "\nProvisioning complete:  Application will start now\n\n"
+}
+function provisioning_has_valid_hf_token() {
+    [[ -n "$HF_TOKEN" ]] || return 1
+    url="https://huggingface.co/api/whoami-v2"
+    response=$(curl -o /dev/null -s -w "%{http_code}" -X GET "$url" \
+        -H "Authorization: Bearer $HF_TOKEN" \
+        -H "Content-Type: application/json")
+    # Check if the token is valid
+    if [ "$response" -eq 200 ]; then
+        return 0
+    else
+        return 1
+    fi
+}
+function provisioning_has_valid_civitai_token() {
+    [[ -n "$CIVITAI_TOKEN" ]] || return 1
+    url="https://civitai.com/api/v1/models?hidden=1&limit=1"
+    response=$(curl -o /dev/null -s -w "%{http_code}" -X GET "$url" \
+        -H "Authorization: Bearer $CIVITAI_TOKEN" \
+        -H "Content-Type: application/json")
+    # Check if the token is valid
+    if [ "$response" -eq 200 ]; then
+        return 0
+    else
+        return 1
+    fi
+}
+# Download from $1 URL to $2 file path
+function provisioning_download() {
+    if [[ -n $HF_TOKEN && $1 =~ ^https://([a-zA-Z0-9_-]+\.)?huggingface\.co(/|$|\?) ]]; then
+        auth_token="$HF_TOKEN"
+    elif
+        [[ -n $CIVITAI_TOKEN && $1 =~ ^https://([a-zA-Z0-9_-]+\.)?civitai\.com(/|$|\?) ]]; then
+        auth_token="$CIVITAI_TOKEN"
+    fi
+    if [[ -n $auth_token ]];then
+        wget --header="Authorization: Bearer $auth_token" -qnc --content-disposition --show-progress -e dotbytes="${3:-4M}" -P "$2" "$1"
+    else
+        wget -qnc --content-disposition --show-progress -e dotbytes="${3:-4M}" -P "$2" "$1"
+    fi
+}
+# Allow user to disable provisioning if they started with a script they didn't want
+if [[ ! -f /.noprovisioning ]]; then
+    provisioning_start
+fi
+# ----------------- CUSTOM NODE AND PROMPT FOLDER SETUP -----------------
+# 1. Ensure prompt folder exists
+mkdir -p /workspace/prompts
+# 2. Add custom PromptFilePicker node
+cat << 'EOF' > /workspace/ComfyUI/custom_nodes/custom_prompt_picker.py
+import os
+PROMPT_DIR = "/workspace/prompts"
+class PromptFilePicker:
+    @classmethod
+    def INPUT_TYPES(cls):
+        try:
+            files = [
+                f for f in os.listdir(PROMPT_DIR)
+                if f.endswith(".txt") and os.path.isfile(os.path.join(PROMPT_DIR, f))
+            ]
+        except Exception as e:
+            files = ["[ERROR: " + str(e) + "]"]
+        files.sort()
+        return {
+            "required": {
+                "prompt_file": (
+                    files,
+                    {"default": files[0]} if files else {}
+                )
+            }
+        }
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("prompt_text",)
+    FUNCTION = "load_prompt"
+    CATEGORY = "Custom/Prompt"
+    def load_prompt(self, prompt_file):
+        path = os.path.join(PROMPT_DIR, prompt_file)
+        with open(path, "r", encoding="utf-8") as f:
+            content = f.read()
+        return (content,)
+NODE_CLASS_MAPPINGS = {
+    "PromptFilePicker": PromptFilePicker
+}
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "PromptFilePicker": "Prompt File Picker"
+}
+EOF
+# ----------------- PATCH START -----------------
+# Force ComfyUI Manager to run with --allow-custom-nodes
+echo '#!/bin/bash
+source /venv/main/bin/activate
+cd /workspace/ComfyUI
+python3 manager.py --allow-custom-nodes' > /workspace/start-comfy-manager.sh
+chmod +x /workspace/start-comfy-manager.sh
+# Replace entrypoint.sh with a version that starts both manager and UI
+echo '#!/bin/bash
+/workspace/start-comfy-manager.sh &  # Launch Manager in background
+cd /workspace/ComfyUI
+python3 main.py --disable-auto-launch --port 18188 --enable-cors-header' > /entrypoint.sh
+chmod +x /entrypoint.sh