LLM-Tuning icon indicating copy to clipboard operation
LLM-Tuning copied to clipboard

lora tuning 出的权重,再加一个合并的功能?

Open litetoooooom opened this issue 2 years ago • 1 comments

作者好

目前用 model = PeftModel.from_pretrained(model, "weights/simple_math_4op").half() 加载Lora,能否提供一个功能merge一下了

litetoooooom avatar Jul 07 '23 08:07 litetoooooom

PefModel类中有一个merge_and_unload函数,可以先试试。

beyondguo avatar Jul 08 '23 02:07 beyondguo