fix #1228

2023-10-19 15:54:10 +08:00 · 2023-10-19 15:54:10 +08:00 · cb0edd2302
parent 6496a99b7d
commit cb0edd2302
1 changed files with 1 additions and 1 deletions
--- 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"):