MMaDA/script/eval_s2t_merging.sh

#!/bin/bash

BASE_DIR="/home/work/AIDAS/ckpts/merged_model"

merge_types=("average_merge_alpha" "hf_common_merge_alpha" "no_vocab_merge_alpha")

alphas=(0.0 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9 1.0)

for merge_type in "${merge_types[@]}"; do
  for alpha in "${alphas[@]}"; do
    CKPT_PATH="${BASE_DIR}/${merge_type}_${alpha}"

    echo "=========================================================="
    echo "Running with ckpt_path=${CKPT_PATH}"
    echo "=========================================================="

    torchrun --nproc_per_node=8 MMaDA/inference_s2t_emova.py \
      config=MMaDA/configs/mmada_demo_s2t.yaml \
      --ckpt_path "${CKPT_PATH}" \
      --train_step 0 \
      --remasking "low_confidence" \
      --generation_step 128 \
      --new_tok 128 \
      --block_length 64
  done
done

echo "Grid search finished."