Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -104,6 +104,13 @@ class WhisperBase(ABC):
         start_time = datetime.now()
         params = WhisperParameters.as_value(*whisper_params)
 
+        # Get the offload params
+        default_params = load_yaml(DEFAULT_PARAMETERS_CONFIG_PATH)        
+        whisper_params = default_params["whisper"]
+        diarization_params = default_params["diarization"]
+        bool_whisper_enable_offload = whisper_params["enable_offload"]
+        bool_diarization_enable_offload = diarization_params["enable_offload"]
+        
         if params.lang is None:
             pass
         elif params.lang == "Automatic Detection":
@@ -166,7 +173,7 @@ class WhisperBase(ABC):
             progress,
             *astuple(params)
         )
-        if params.whisper_enable_offload:
+        if bool_whisper_enable_offload:
             self.offload()
 
         if params.vad_filter:
@@ -187,7 +194,7 @@ class WhisperBase(ABC):
                 transcribed_result=result,
                 device=params.diarization_device
             )
-            if params.diarization_enable_offload:
+            if bool_diarization_enable_offload:
                 self.diarizer.offload()
 
         self.cache_parameters(