utils/apply_lora.py 有个小Bug
Closed this issue · 1 comments
apachemycat commented
apply_lora(args.base_model_path, args.target_model_path, args.lora_path,
args.save_target_model)
def apply_lora(
base_model_path: str,
lora_path: str,
load_8bit: bool = False,
target_model_path: str = None,
save_target_model: bool = False
) -> Tuple[AutoModelForCausalLM, AutoTokenizer]:
两个变量的位置错了
jianzhnie commented
非常感谢，bug 已修复