Former-commit-id: 67af68f4fc5232760c57b3a0ae780628da09db6a
This commit is contained in:
hiyouga
2024-06-03 23:30:37 +08:00
parent 4c1f015eca
commit 26bc79f971
5 changed files with 26 additions and 13 deletions

View File

@@ -64,10 +64,15 @@ class Runner:
return ALERTS["err_demo"][lang]
if do_train:
if not get("train.output_dir"):
return ALERTS["err_no_output_dir"][lang]
stage = TRAINING_STAGES[get("train.training_stage")]
reward_model = get("train.reward_model")
if stage == "ppo" and not reward_model:
if stage == "ppo" and not get("train.reward_model"):
return ALERTS["err_no_reward_model"][lang]
else:
if not get("eval.output_dir"):
return ALERTS["err_no_output_dir"][lang]
if not from_preview and not is_gpu_or_npu_available():
gr.Warning(ALERTS["warn_no_cuda"][lang])