cmedina/agenx-lora-training
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: mixed_precision -> bf16/api change

Browse Source
This commit is contained in:
Christian Medina
2026-06-30 20:34:05 -04:00
parent 9280103260
commit 9d5d4eebc7
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
training/scripts/train.py
View File

@@ -89,7 +89,8 @@ def train(config_path):
save_total_limit=config["train_params"]["save_total_limit"],
eval_strategy=config.get("eval_strategy", "steps"),
eval_steps=config.get("eval_steps", 100),
mixed_precision=config.get("mixed_precision", "bf16"),
bf16=config.get("mixed_precision", "bf16") == "bf16",
fp16=config.get("mixed_precision", "bf16") == "fp16",
gradient_checkpointing=config.get("gradient_checkpointing", True),
)