From 243223d89949b449335167e7bc06b97f0779e528 Mon Sep 17 00:00:00 2001 From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com> Date: Thu, 2 Jul 2026 11:38:37 -0400 Subject: [PATCH] fix: fix indentation error in strategy 3 --- train.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/train.py b/train.py index 79390e4..98ecf39 100644 --- a/train.py +++ b/train.py @@ -89,15 +89,15 @@ def train(config_path): # -------------------------------------------------------------- print("\n[3/4] Trying: bf16 CPU...") try: - model = AutoModelForCausalLM.from_pretrained( - config["base_model"], - torch_dtype=torch.bfloat16, - device_map="cpu", - low_cpu_mem_usage=True, - trust_remote_code=True, - ) - print("✓ Success: bf16 CPU") - except Exception as e: + model = AutoModelForCausalLM.from_pretrained( + config["base_model"], + torch_dtype=torch.bfloat16, + device_map="cpu", + low_cpu_mem_usage=True, + trust_remote_code=True, + ) + print("✓ Success: bf16 CPU") + except Exception as e: errors.append(("bf16 CPU", e)) print(f"✗ Failed: {e}")