
LLM_path="" #TODO: LLM path
data_path="" #TODO: training dataset path
output_dir="" # TODO: CCoT output dir


############################################
#! 1. Optimize for generating CCoT
############################################
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python optimize.py \
    --verbose False \
    --data_path "$data_path" \
    --exist_index_path "" \
    --Qwen_model_path "$LLM_path" \
    --output_embedding_trace_dir "$output_dir" \
    --max_step 128 \
    --loop_step 8 \
    --ccot_length 512 \
    --n_thread 8


