MMaDA/inference/run_t2s_grid_timesteps.sh

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

# Grid runner: sweep TIMESTEPS over powers of two (2..256)

# Default grid (space-separated). Override with TIMESTEPS_GRID="2 4 ..." if needed.
TIMESTEPS_GRID=${TIMESTEPS_GRID:-"2 4 8 16 32 64 128 256"}

# Pass all KEY=VALUE and CLI flags through to the base runner.
REST_ARGS=("$@")

for TS in $TIMESTEPS_GRID; do
  echo "[t2s-grid] Running TIMESTEPS=$TS"
  bash MMaDA/inference/run_t2s.sh TIMESTEPS="$TS" "${REST_ARGS[@]}"
done

# Examples:
  # bash MMaDA/inference/run_t2s_grid_timesteps.sh \
  #   WER=1 ASR=openai/whisper-large-v3 LANG=english TEXT_NORM=basic WER_SAMPLES=128 \
  #   MODE=free GUIDANCE_SCALE=1.5 SEQ_LEN=512 BATCH_SIZE=1 \
  #   OUTPUT_DIR=inference/outputs/t2s_cli \
  #   TRAIN_CONFIG=MMaDA/configs/omada_pretraining_stage1-3.yaml \
  #   CKPT_ROOT=ckpts/omada/omada-training-stage1_4th/checkpoint-330000/unwrapped_model