diff --git a/src/llmtuner/webui/runner.py b/src/llmtuner/webui/runner.py index 9fde4d2c..f951fe2c 100644 --- a/src/llmtuner/webui/runner.py +++ b/src/llmtuner/webui/runner.py @@ -179,7 +179,7 @@ class Runner: max_new_tokens=get("eval.max_new_tokens"), top_p=get("eval.top_p"), temperature=get("eval.temperature"), - output_dir=get("eval.output_dir") + output_dir=output_dir ) if get("eval.predict"):