#!/bin/bash

llms=()

N=2
M=2
O=2

llm="azure_4omini_registry"
PROMPT="prompts/box-prompt-v2/dmas-think-qwen-fullplan.py"
ENV="box1"
DATA="data_cache/robot/box1-small-test"
run_name="prompt-v2"

llms=(
    # "azure_4omini_registry"
    # "qwen_registry"
    "qwen3-4b_registry"
    # "mistral_registry"
    # "qwq_registry"
    # "r1-qwen-1.5b_registry"
    # "r1gpt@registry"
    # "grok-qwq@registry"
)
RUNNER="fullplan"

export ALL_GPU=True
for llm in "${llms[@]}"; do
    python scripts/run_box.py \
        data=$DATA \
        env_name=$ENV \
        env_config.grid_N=$N \
        env_config.grid_M=$M \
        env_config.num_objects=$O \
        prompt_file=$PROMPT \
        llm_name=$llm \
        runner=$RUNNER \
        sample_params.temperature=0.7 \
        sample_params.top_p=0.95 \
        sample_params.max_tokens=8192 \
        sample_params.n=4 \
        run_name=$run_name
done