LLM-Tuning
LLM-Tuning copied to clipboard
lora tuning 出的权重,再加一个合并的功能?
作者好
目前用 model = PeftModel.from_pretrained(model, "weights/simple_math_4op").half() 加载Lora,能否提供一个功能merge一下了
PefModel类中有一个merge_and_unload函数,可以先试试。