LaVieEnRose365 / ReLLa

Code of Paper "ReLLa: Retrieval-enhanced Large Language Models for Mitigating Long Context Problems in Recommendation".
41 stars 4 forks source link

请问在代码中如何修改设置,来分别执行zero_shot实验和few_shot实验? #11

Open Cxx-0 opened 4 months ago

LaVieEnRose365 commented 2 months ago

直接在inference.py中就可以,zero_shot是加载原始模型,few_shot即finetune后的模型,需要拼上lora adapter,即use_lora=True