fix bug in web ui

This commit is contained in:
hiyouga committed 2023-11-16 15:21:24 +08:00
1 parent 7537dd434f
commit 6efa38be46
3 files changed
+19 -13

No files matched your search

-1
View File
@@ -202,7 +202,6 @@ def load_model_and_tokenizer(
# Prepare model with valuehead for RLHF
if stage in ["rm", "ppo"]:
model: "AutoModelForCausalLMWithValueHead" = AutoModelForCausalLMWithValueHead.from_pretrained(model)
reset_logging()
vhead_path = (
model_args.checkpoint_dir[-1] if model_args.checkpoint_dir is not None else model_args.model_name_or_path
)