OpenMOSS/CoLLiE

No module named 'collie.callbacks.pefts'

Closed this issue · 2 comments

跑llama finetune lora报错找不到collie.callbacks.pefts,请问LoRACallback的实现在哪?
Traceback (most recent call last):
File "finetune_llama_lora.py", line 27, in
from collie.callbacks.pefts.lora import LoRACallback
ModuleNotFoundError: No module named 'collie.callbacks.pefts'

你好,没有loracallback这个类,使用lora可以参考https://github.com/OpenLMLab/collie/blob/dev/examples/peft/finetune_internlm_for_classification.py

多谢啦