From 7d66c9535ca68eb0d2ddcee427611aed104a89f8 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 30 Apr 2025 10:07:38 -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 143b8c68a2..56499cec30 100644 --- a/unsloth/dataprep/synthetic.py +++ b/unsloth/dataprep/synthetic.py @@ -100,8 +100,6 @@ class SyntheticDataKit: start_new_session = True, ) self.vllm_process = vllm_process - atexit.register(self.destroy_vllm) - self._finalizer = weakref.finalize(self, self.destroy_vllm) ready_message_part = b"Starting vLLM API server on" ready = False @@ -192,7 +190,9 @@ class SyntheticDataKit: def __enter__(self): return self def __exit__(self, *exc): self.destroy_vllm() - def __del__(self): self.destroy_vllm() + def __del__(self): + print("In del") + self.destroy_vllm() def truncate(self, filename = None): # Truncates by summary and max generation