File size: 455 Bytes
936459a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | 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 |