Update synthetic.py

2025-05-14 04:05:23 -07:00 · 2025-05-14 04:05:23 -07:00 · 0bbf131238
commit 0bbf131238
parent f971fce721
1 changed files with 1 additions and 5 deletions
--- a/unsloth/dataprep/synthetic.py
+++ b/unsloth/dataprep/synthetic.py
@ -196,11 +196,7 @@ class SyntheticDataKit:
            gc.collect()

        # Delete vLLM module as well
-        # We delete llm.llm_engine.model_executor, so first make it accessible
-        class Dummy0: model_executor = 1
-        class Dummy1: llm_engine = Dummy0()
-        class Dummy2: llm = Dummy1()
-        delete_vllm(Dummy2().llm)
+        delete_vllm(llm = None)
    pass

    def __enter__(self): return self