Update generative-proof-of-concept-CPU-preprocessing-in-memory.py

david-thrower · web-flow · commit 694c5618514d · 2025-09-18T02:38:56.000-04:00
Add garbage collection after run.
diff --git a/generative-proof-of-concept-CPU-preprocessing-in-memory.py b/generative-proof-of-concept-CPU-preprocessing-in-memory.py
@@ -940,6 +940,9 @@ def complete_text(text):
             print(f"PROMPT number {counter}: {half_sample}; RESPONSE: {full_generated_text}")
             counter += 1
         mlflow.log_metric("perplexity", result, step=trial.number)
+        del(best_model_found)
+        del(generator)
+        collect()
         return result            
 
 def main():
