diff --git a/unsloth/dataprep/synthetic.py b/unsloth/dataprep/synthetic.py
index a2c2e0cf07..45805bbfb8 100644
--- a/unsloth/dataprep/synthetic.py
+++ b/unsloth/dataprep/synthetic.py
@@ -81,6 +81,13 @@ class SyntheticDataKit:
         print(engine_args)
         if "dtype" in engine_args:
             dtype_val = engine_args["dtype"]
+            dtype_mapping = {
+                torch.float16  : "float16",
+                torch.bfloat16 : "bfloat16",
+                torch.float32  : "float32",
+            }
+            if dtype_val in dtype_mapping:
+                dtype_val = dtype_mapping[dtype_val]
             # Convert torch.bfloat16, torch.float16, etc. to valid CLI string
             if hasattr(dtype_val, "name"):
                 engine_args["dtype"] = dtype_val.name