export CUDA_VISIBLE_DEVICES=0,1,2,3

python script/data/slm_prefill.py \
    --num_gpu 4 \
    --dataset_path ./path/to/dataset.jsonl \
    --test_model_list ./path/to/model  \
    --output_path ./path/to/output \
    --max_input_length 40000 \
    --temperature 0.0 \
    --top_p 1.0