diff --git a/Install_CorridorKey_Linux_Mac.sh b/Install_CorridorKey_Linux_Mac.sh index ce1699d37..26ae09759 100755 --- a/Install_CorridorKey_Linux_Mac.sh +++ b/Install_CorridorKey_Linux_Mac.sh @@ -84,6 +84,24 @@ else fi fi +# CorridorKeyBlue (dedicated blue-screen weights). Optional: the green +# checkpoint above is the primary install. If the blue download fails — repo +# missing, network blip, etc. — we just skip it. The Python runtime +# (_ensure_torch_checkpoint) will retry the download on first --screen-color +# blue use, so a failure here only costs the user a one-time 300 MB pull at +# render time instead of bricking the install. +BLUE_SAFETENSORS_PATH="$CKPT_DIR/CorridorKeyBlue_1.0.safetensors" +HF_BLUE_BASE="https://huggingface.co/nikopueringer/CorridorKeyBlue_1.0/resolve/main" +if [ -f "$BLUE_SAFETENSORS_PATH" ]; then + echo "CorridorKeyBlue checkpoint already exists!" +else + echo "Downloading CorridorKeyBlue_1.0.safetensors (blue-screen model)..." + if ! curl -L --fail -o "$BLUE_SAFETENSORS_PATH" "$HF_BLUE_BASE/CorridorKeyBlue_1.0.safetensors"; then + echo "[INFO] Blue checkpoint not downloaded — it will fetch automatically the first time you run with --screen-color blue." + rm -f "$BLUE_SAFETENSORS_PATH" + fi +fi + echo "" echo "===================================================" echo " Setup Complete! You are ready to key!" diff --git a/Install_CorridorKey_Windows.bat b/Install_CorridorKey_Windows.bat index 7708d5053..cc584ad57 100644 --- a/Install_CorridorKey_Windows.bat +++ b/Install_CorridorKey_Windows.bat @@ -70,6 +70,23 @@ if exist "%SAFETENSORS_PATH%" ( ) ) +REM CorridorKeyBlue (dedicated blue-screen weights). Optional download: +REM if it fails (repo missing, network blip, etc.) the Python runtime will +REM retry on first --screen-color blue use, so failure here is non-fatal -- +REM users keying green plates aren't blocked by an unrelated download. +set "BLUE_SAFETENSORS_PATH=%CKPT_DIR%\CorridorKeyBlue_1.0.safetensors" +set "HF_BLUE_BASE=https://huggingface.co/nikopueringer/CorridorKeyBlue_1.0/resolve/main" +if exist "%BLUE_SAFETENSORS_PATH%" ( + echo CorridorKeyBlue checkpoint already exists! +) else ( + echo Downloading CorridorKeyBlue_1.0.safetensors ^(blue-screen model^)... + curl.exe -L --fail -o "%BLUE_SAFETENSORS_PATH%" "%HF_BLUE_BASE%/CorridorKeyBlue_1.0.safetensors" + if errorlevel 1 ( + echo [INFO] Blue checkpoint not downloaded -- it will fetch automatically the first time you run with --screen-color blue. + if exist "%BLUE_SAFETENSORS_PATH%" del "%BLUE_SAFETENSORS_PATH%" + ) +) + echo. echo =================================================== echo Setup Complete! You are ready to key!