Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -106,11 +106,13 @@ class WhisperBase(ABC):
         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"]
+        #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"]
+        bool_whisper_enable_offload = True
+        bool_diarization_enable_offload = True
         
         if params.lang is None:
             pass