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 |
|