fix webui

Former-commit-id: 2c8b7414be9b43e20cc1d0575cc4dc1c7545fd86
2023-08-12 23:52:07 +08:00
parent a22982f2fa
commit 6c7225a5d4
3 changed files with 3 additions and 5 deletions
--- a/src/llmtuner/webui/runner.py
+++ b/src/llmtuner/webui/runner.py
@@ -126,8 +126,6 @@ class Runner:
            logging_steps=logging_steps,
            save_steps=save_steps,
            warmup_steps=warmup_steps,
-            fp16=(compute_type == "fp16"),
-            bf16=(compute_type == "bf16"),
            padding_side=padding_side,
            lora_rank=lora_rank,
            lora_dropout=lora_dropout,
@@ -135,7 +133,7 @@ class Runner:
            resume_lora_training=resume_lora_training,
            output_dir=output_dir
        )
-
+        args[compute_type] = True
        if val_size > 1e-6:
            args["val_size"] = val_size
            args["evaluation_strategy"] = "steps"