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

fix: fix indentation error in strategy 3

Browse Source
This commit is contained in:
Christian Medina
2026-07-02 11:38:37 -04:00
parent 1b3f678b50
commit 243223d899
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
train.py
View File

@@ -89,15 +89,15 @@ def train(config_path):
# -------------------------------------------------------------- # --------------------------------------------------------------
print("\n[3/4] Trying: bf16 CPU...") print("\n[3/4] Trying: bf16 CPU...")
try: try:
model = AutoModelForCausalLM.from_pretrained( model = AutoModelForCausalLM.from_pretrained(
config["base_model"], config["base_model"],
torch_dtype=torch.bfloat16, torch_dtype=torch.bfloat16,
device_map="cpu", device_map="cpu",
low_cpu_mem_usage=True, low_cpu_mem_usage=True,
trust_remote_code=True, trust_remote_code=True,
) )
print("✓ Success: bf16 CPU") print("✓ Success: bf16 CPU")
except Exception as e: except Exception as e:
errors.append(("bf16 CPU", e)) errors.append(("bf16 CPU", e))
print(f"✗ Failed: {e}") print(f"✗ Failed: {e}")