#!/bin/bash

export CUDA_VISIBLE_DEVICES=1,2,3

python test.py \
  --checkpoint_path /home/dataset-assist-0/usr/lh/ysh/dw/RL/AR/checkpoints/20260614_235422/model_epoch_7.pth \
  --archive_path /home/dataset-assist-0/usr/lh/ysh/dw/RL/AR/data/shards/shard_0000.tar.zst \
  --num_refine_groups 3 \
  --model_device cuda:0 \
  --vae_device cuda:0 \
  --clip_device cuda:0