From be7a7636b771c304a7df2f877ec1a0165a14ff05 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 30 Apr 2025 10:22:31 -0700 Subject: [PATCH] Update synthetic.py --- unsloth/dataprep/synthetic.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/unsloth/dataprep/synthetic.py b/unsloth/dataprep/synthetic.py index a49a312a9d..5aa9229597 100644 --- a/unsloth/dataprep/synthetic.py +++ b/unsloth/dataprep/synthetic.py @@ -157,7 +157,7 @@ class SyntheticDataKit: pass pass - def destroy_vllm(self): + def cleanup(self): if not hasattr(self, "vllm_process"): return vllm_process = self.vllm_process @@ -187,8 +187,8 @@ class SyntheticDataKit: pass def __enter__(self): return self - def __exit__(self, *exc): self.destroy_vllm() - def __del__(self): self.destroy_vllm() + def __exit__(self, *exc): self.cleanup() + def __del__(self): self.cleanup() def truncate(self, filename = None): # Truncates by summary and max generation