Benchmark-Single / infer.sh
Junyin's picture
Add files using upload-large-folder tool
811e03d verified
raw
history blame contribute delete
353 Bytes
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