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 10:22:31 -07:00
commit be7a7636b7
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/dataprep/synthetic.py
View file

@ -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