ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update synthetic.py

Browse source
This commit is contained in:
Daniel Han 2025-07-19 03:11:33 -07:00
commit 6a65ee478c
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
unsloth/dataprep/synthetic.py
View file

@ -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