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-04-30 00:00:09 -07:00
commit 8f9dc57b29
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
unsloth/dataprep/synthetic.py
View file

@ -71,9 +71,12 @@ def async_load_vllm(
for key, value in engine_args.items():
flag = "--" + key.replace("_", "-")
which = str(value).lower().replace("torch.", "")
if which == "true" or which == "false":
# Ignore --enforce-eager True / False
if which == "true":
# Ignore --enforce-eager True
subprocess_commands += [flag,]
elif which == "false":
# Add --no-enforce-eager
subprocess_commands += ["no-" + flag,]
else:
subprocess_commands += [flag, which,]
pass
@ -254,7 +257,7 @@ def configure_synthetic_data_kit(
for path in locations:
os.makedirs(os.path.join(output_folder, path), exist_ok = True)
pass
config = synthetic_config_string\
.replace("{model_name}", str(model_name))\
.replace("{temperature}", str(temperature))\