From 1fe9fa59f6646a00fd7b89f1d4a989bbd9c366dd Mon Sep 17 00:00:00 2001
From: McCloudS <64094529+McCloudS@users.noreply.github.com>
Date: Sat, 13 Jul 2024 13:21:43 -0600
Subject: [PATCH] Fix garbage collection

Do garbage collection after every task.
---
 subgen.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/subgen.py b/subgen.py
index 5019e64..57624fe 100644
--- a/subgen.py
+++ b/subgen.py
@@ -504,11 +504,11 @@ def start_model():
         model = stable_whisper.load_faster_whisper(whisper_model, download_root=model_location, device=transcribe_device, cpu_threads=whisper_threads, num_workers=concurrent_transcriptions, compute_type=compute_type)
 
 def delete_model():
+    gc.collect()
     if clear_vram_on_complete and task_queue.qsize() == 0:
         global model
         logging.debug("Queue is empty, clearing/releasing VRAM")
         model = None
-        gc.collect()
 
 def isAudioFileExtension(file_extension):
     return file_extension.casefold() in \