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"]
+        print("offload whisper param: "+whisper_params["enable_offload"])
+        print("offload diarization param: "+diarization_params["enable_offload"])
         bool_whisper_enable_offload = True
         bool_diarization_enable_offload = True