Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -111,10 +111,6 @@ class WhisperBase(ABC):
|
|
111 |
diarization_params = default_params["diarization"]
|
112 |
bool_whisper_enable_offload = whisper_params["enable_offload"]
|
113 |
bool_diarization_enable_offload = diarization_params["enable_offload"]
|
114 |
-
print("offload whisper param: "+str(whisper_params["enable_offload"]))
|
115 |
-
print("offload diarization param: "+str(diarization_params["enable_offload"]))
|
116 |
-
#bool_whisper_enable_offload = True
|
117 |
-
#bool_diarization_enable_offload = True
|
118 |
|
119 |
if params.lang is None:
|
120 |
pass
|
@@ -391,8 +387,8 @@ class WhisperBase(ABC):
|
|
391 |
|
392 |
return [result_str,result_file_path,total_info]
|
393 |
|
394 |
-
except Exception as e:
|
395 |
-
print(f"Error transcribing file: {e}")
|
396 |
finally:
|
397 |
self.release_cuda_memory()
|
398 |
|
|
|
111 |
diarization_params = default_params["diarization"]
|
112 |
bool_whisper_enable_offload = whisper_params["enable_offload"]
|
113 |
bool_diarization_enable_offload = diarization_params["enable_offload"]
|
|
|
|
|
|
|
|
|
114 |
|
115 |
if params.lang is None:
|
116 |
pass
|
|
|
387 |
|
388 |
return [result_str,result_file_path,total_info]
|
389 |
|
390 |
+
# except Exception as e:
|
391 |
+
# print(f"Error transcribing file: {e}")
|
392 |
finally:
|
393 |
self.release_cuda_memory()
|
394 |
|