Update synthetic.py

This commit is contained in:
Daniel Han 2025-07-19 02:54:28 -07:00
commit 4f73de8041

View file

@ -78,6 +78,7 @@ class SyntheticDataKit:
use_bitsandbytes = False,
**kwargs,
)
print(engine_args)
if "dtype" in engine_args:
dtype_val = engine_args["dtype"]
# Convert torch.bfloat16, torch.float16, etc. to valid CLI string
@ -113,6 +114,7 @@ class SyntheticDataKit:
else:
subprocess_commands += ["--" + flag, which,]
pass
print(subprocess_commands)
vllm_process = subprocess.Popen(
subprocess_commands,
stdout = subprocess.PIPE,