From 5b8b1e9b763b1a363f422d86efd48cbc81509a2a Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 30 Apr 2025 00:03:54 -0700 Subject: [PATCH] Update synthetic.py --- unsloth/dataprep/synthetic.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/unsloth/dataprep/synthetic.py b/unsloth/dataprep/synthetic.py index 4dae6cba4c..ad4bdfcc5f 100644 --- a/unsloth/dataprep/synthetic.py +++ b/unsloth/dataprep/synthetic.py @@ -69,18 +69,17 @@ def async_load_vllm( "vllm", "serve", str(model_name), ] for key, value in engine_args.items(): - flag = "--" + key.replace("_", "-") + flag = key.replace("_", "-") which = str(value).lower().replace("torch.", "") if which == "true": # Ignore --enforce-eager True - subprocess_commands += [flag,] + subprocess_commands += ["--" + flag,] elif which == "false": # Add --no-enforce-eager - subprocess_commands += ["no-" + flag,] + subprocess_commands += ["--no-" + flag,] else: - subprocess_commands += [flag, which,] + subprocess_commands += ["--" + flag, which,] pass - print(subprocess_commands) vllm_process = subprocess.Popen( subprocess_commands, stdout = subprocess.PIPE,