423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-15 19:30:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix webui

Browse Source
This commit is contained in:
hiyouga
2023-08-12 23:52:07 +08:00
parent 9dc6a296e3
commit d69b1388e6
3 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/webui/runner.py
View File

@@ -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"