pip install -r requirements.txt
python rec.py \
--base_model 'decapoda-research/llama-7b-hf' \
--data_path './moives' \
--output_dir './checkpoint'
python generate.py \
--load_8bit \
--base_model 'decapoda-research/llama-7b-hf' \
--lora_weights './checkpoint/movies'
This project is implemented based on alpaca-lora (https://github.com/tloen/alpaca-lora)
@article{ji2024genrec,
title={GenRec: Large Language Model for Generative Recommendation},
author={Jianchao Ji, Zelong Li, Shuyuan Xu, Wenyue Hua, Yingqiang Ge, Juntao Tan and Yongfeng Zhang},
journal={ECIR},
year={2024}
}