File size: 2,380 Bytes
406e545 1eb7607 406e545 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
cat > /workspace/bootstrap_forge.sh <<'EOF'
#!/usr/bin/env bash
set -euo pipefail
# 1) caches persistentes
mkdir -p /workspace/.cache/pip /workspace/.cache/huggingface
cat > /workspace/env.sh <<'EOS'
export PIP_CACHE_DIR=/workspace/.cache/pip
export HF_HOME=/workspace/.cache/huggingface
export HUGGINGFACE_HUB_CACHE=/workspace/.cache/huggingface/hub
EOS
# carrega env persistente (não repita exports soltos)
source /workspace/env.sh
# 2) repo (idempotente)
REPO_DIR="/workspace/stable-diffusion-webui-forge"
if [ ! -d "$REPO_DIR/.git" ]; then
git clone https://github.com/lllyasviel/stable-diffusion-webui-forge.git "$REPO_DIR"
else
git -C "$REPO_DIR" fetch --all --prune
git -C "$REPO_DIR" pull --ff-only
fi
# 3) venv persistente (herda torch do template)
if [ ! -d /workspace/venv-hh ]; then
python3.10 -m venv --system-site-packages /workspace/venv-hh
fi
source /workspace/venv-hh/bin/activate
python -m pip install -U pip wheel
# 4) deps de sistema (necessárias para pycairo)
export DEBIAN_FRONTEND=noninteractive
apt-get update
apt-get install -y --no-install-recommends pkg-config libcairo2-dev python3.10-dev cmake
rm -rf /var/lib/apt/lists/*
# 5) deps Python que te deram erro antes
python -m pip install -U joblib gdown
python -m pip install -U pycairo svglib
# 6) requirements do Forge (sem forçar reinstalar torch)
cd "$REPO_DIR"
grep -vE '^(torch|torchvision|torchaudio)\b' requirements_versions.txt > /workspace/req-no-torch.txt
python -m pip install -r /workspace/req-no-torch.txt
# 7) fixa o venv_dir de forma persistente para o webui.sh
cat > "$REPO_DIR/webui-user.sh" <<'EOW'
venv_dir="/workspace/venv-hh"
EOW
mkdir /workspace/stable-diffusion-webui-forge/models/ESRGAN
cd /workspace/stable-diffusion-webui-forge/models/ESRGAN
wget https://huggingface.co/willseijits/HH12Solo80/resolve/main/4xUltrasharp_4xUltrasharpV10.pth
# 8) cria um "starter" para reinícios (ver abaixo)
cat > /workspace/start_forge.sh <<'EOS'
#!/usr/bin/env bash
set -euo pipefail
source /workspace/env.sh
source /workspace/venv-hh/bin/activate
cd /workspace/stable-diffusion-webui-forge
# Ajuste flags conforme seu risco/necessidade:
./webui.sh -f --listen --port 7860 --enable-insecure-extension-access
EOS
chmod +x /workspace/start_forge.sh
echo "OK. Para iniciar (ou reiniciar) o Forge: bash /workspace/start_forge.sh"
EOF
bash /workspace/bootstrap_forge.sh
|