Merge pull request #134 from rmusser01/dev

Update Audio_Transcription_Lib.py
rmusser01 · Jul 28, 2024 · a288875 · a288875
2 parents b6535be + c14f8b5
commit a288875
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/App_Function_Libraries/Audio_Transcription_Lib.py b/App_Function_Libraries/Audio_Transcription_Lib.py
@@ -45,9 +45,11 @@
 processing_choice = config.get('Processing', 'processing_choice', fallback='cpu')
 
 
+# FIXME: This is a temporary solution.
+# This doesn't clear older models, which means potentially a lot of memory is being used...
 def get_whisper_model(model_name, device):
     global whisper_model_instance
-    if whisper_model_instance is None or whisper_model_instance.model_size != model_name:
+    if whisper_model_instance is None:
         from faster_whisper import WhisperModel
         logging.info(f"Initializing new WhisperModel with size {model_name} on device {device}")
         whisper_model_instance = WhisperModel(model_name, device=device)