From f0ee6bc9a2536fa80804e25acaf2a9bbc98c1a93 Mon Sep 17 00:00:00 2001
From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com>
Date: Wed, 1 Jul 2026 07:38:36 -0400
Subject: [PATCH] fix: torch_dtype -> dtype (fix deprecation warning)

---
 training/scripts/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/training/scripts/train.py b/training/scripts/train.py
index 627d90c..0462e42 100755
--- a/training/scripts/train.py
+++ b/training/scripts/train.py
@@ -36,7 +36,7 @@ def train(config_path):
     print(f"Loading model: {config['base_model']}")
     model = AutoModelForCausalLM.from_pretrained(
         config["base_model"],
-        torch_dtype=torch.bfloat16,  # Convert FP8 -> bf16
+        dtype=torch.bfloat16,  # Convert FP8 -> bf16
         device_map="cpu",  # Load to CPU first
         trust_remote_code=True,
     )