run_hpsv2.sh

#!/usr/bin/env bash
set -euo pipefail

# ─────────────── настраиваемые переменные (или задайте их при вызове):
NUM_GPUS=3   # сколько GPU использовать
STEPS=${STEPS:-40}                               # num_inference_steps
GSCALE=${GSCALE:-4.5}                            # guidance scale (CFG)
DTYPE=${DTYPE:-bf16}                             # bf16 | fp16
SEED=${SEED:-42}                                 # базовый сид
HPS_VERSION=${HPS_VERSION:-v2.1}                 # v2 | v2.1
XFORMERS=${XFORMERS:-1}                          # 1 → --xformers, 0 → без

ACCELERATE_EXTRA="${ACCELERATE_EXTRA:-}"

SCRIPT=scripts/run_hpsv2.py
XFORMERS_FLAG=$([ "$XFORMERS" = 1 ] && echo "--xformers" || echo "")
PY_ARGS="--steps $STEPS --gscale $GSCALE --dtype $DTYPE --seed $SEED --hps_version $HPS_VERSION $XFORMERS_FLAG"

CMD="accelerate launch --multi_gpu --num_processes $NUM_GPUS $ACCELERATE_EXTRA $SCRIPT $PY_ARGS"

echo "[$(date +%H:%M:%S)] ⇢ $CMD"
eval $CMD