ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

tiny fix

Browse Source 
Former-commit-id: acc99ef2fb62908288f88369354135d581588b63
This commit is contained in:
hiyouga
2024-02-21 18:30:29 +08:00
parent bc16c9a54a
commit 67deefe527
3 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/webui/runner.py
View File

@@ -144,7 +144,7 @@ class Runner:
args["lora_target"] = get("train.lora_target") or get_module(get("top.model_name"))
args["additional_target"] = get("train.additional_target") or None
args["use_rslora"] = get("train.use_rslora")
if TRAINING_STAGES[get("train.training_stage")] in ["rm", "ppo", "dpo"]:
if args["stage"] in ["rm", "ppo", "dpo"]:
args["create_new_adapter"] = args["quantization_bit"] is None
else:
args["create_new_adapter"] = get("train.create_new_adapter")
@@ -166,7 +166,7 @@ class Runner:
args["val_size"] = get("train.val_size")
args["evaluation_strategy"] = "steps"
args["eval_steps"] = get("train.save_steps")
args["load_best_model_at_end"] = True
args["load_best_model_at_end"] = args["stage"] not in ["rm", "ppo"]
return args