Benchmark-Dual / infer.sh
Junyin's picture
Add files using upload-large-folder tool
936459a verified
CKPT_PATH=${datain}/v-yinju/rq-llama/v6/Instruments
python generate_embeddings.py \
--ckpt_path $CKPT_PATH \
--item_save_path $CKPT_PATH/embeddings.item.tsv \
--user_save_path $CKPT_PATH/embeddings.user.tsv \
--device_map 0
python generate_indices.py \
--ckpt_path $CKPT_PATH \
--item_data_path $CKPT_PATH/embeddings.item.tsv \
--user_data_path $CKPT_PATH/embeddings.user.tsv \
--save_path $CKPT_PATH \
--device_map 0