From 04ec7af3e85612a3706c4e47822dae3aae646902 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sat, 19 Jul 2025 03:36:53 -0700 Subject: [PATCH] Update synthetic.py --- unsloth/dataprep/synthetic.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/dataprep/synthetic.py b/unsloth/dataprep/synthetic.py index 33b158c0cd..94510ea163 100644 --- a/unsloth/dataprep/synthetic.py +++ b/unsloth/dataprep/synthetic.py @@ -84,6 +84,7 @@ class SyntheticDataKit: if dtype_val == torch.float16: dtype_val = "float16" elif dtype_val == torch.bfloat16: dtype_val = "bfloat16" elif dtype_val == torch.float32: dtype_val = "float32" + engine_args["dtype"] = dtype_val # Convert torch.bfloat16, torch.float16, etc. to valid CLI string if hasattr(dtype_val, "name"): engine_args["dtype"] = dtype_val.name