CKPT_PATH=${datain}/v-yinju/rq-llama/v3-train/Instruments/checkpoint-14517

python generate_embeddings.py \
    --ckpt_path $CKPT_PATH \
    --save_path $CKPT_PATH/embeddings.tsv \
    --device_map 0

python generate_indices_v2.py \
    --ckpt_path $CKPT_PATH \
    --data_path $CKPT_PATH/embeddings.tsv \
    --save_path $CKPT_PATH \
    --device_map 0