p-doom/crowd-code-dataset-0.2 huggingface dataset:
[tab] slurm/dev/franz/berlin/crowd-pilot/qwen_0_6/maxtext_decode.sh (shellscript)
#!/usr/bin/env bash

#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH --gres=gpu:1
#SBATCH --time=24:00:00
#SBATCH --cpus-per-task=8
#SBATCH --output=/fast/project/HFMI_SynergyUnit/jafar_ws/logs/franz/crowd-pilot/maxtext/%x_%j.log
#SBATCH --error=/fast/project/HFMI_SynergyUnit/jafar_ws/logs/franz/crowd-pilot/maxtext/%x_%j.log
#SBATCH --job-name=crowd-pilot_qwen3-0.6b_maxtext_decode
#SBATCH --requeue
#SBATCH --signal=b:usr1@300 # 5 min before timeout

# Log the sbatch script
cat $0

cd maxtext
source .venv/bin/activate

CHECKPOINT_PATH=/fast/project/HFMI_SynergyUnit/jafar_ws/data/crowd-pilot/outputs/crowd-pilot_qwen3-0.6b_batch_size_4_32858/checkpoints/0/items
PROMPT=$(cat << 'EOF'
<act focus file="sbatch_scripts/coinrun/train_tokenizer_coinrun.sh" />
// observation
EOF
)
MODEL_NAME=qwen3-0.6b
PER_DEVICE_BATCH_SIZE=1
MAX_TARGET_LENGTH=128

python3 -m MaxText.decode src/MaxText/configs/base.yml\
    load_parameters_path=$CHECKPOINT_PATH\
    tokenizer_path=src/MaxText/assets/qwen3-tokenizer\
    prompt="$PROMPT"\
    model_name=$MODEL_NAME\
    per_device_batch_size=$PER_DEVICE_BATCH_SIZE\
    max_target_length=$MAX_TARGET_LENGTH\
    skip_jax_distributed_system=True