cjones/WhisperLive
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Raise error for invalid model paths

Browse Source
This commit is contained in:
Andreas Peldszus
2024-05-31 14:45:51 +02:00
parent 3c09289dea
commit 3a96f60006
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
whisper_live/server.py
+5 -1
View File
@@ -300,11 +300,15 @@ class TranscriptionServer:
host (str): The host address to bind the server. host (str): The host address to bind the server.
port (int): The port number to bind the server. port (int): The port number to bind the server.
""" """
if faster_whisper_custom_model_path is not None and not os.path.exists(faster_whisper_custom_model_path):
raise ValueError(f"Custom faster_whisper model '{faster_whisper_custom_model_path}' is not a valid path.")
if whisper_tensorrt_path is not None and not os.path.exists(whisper_tensorrt_path):
raise ValueError(f"TensorRT model '{whisper_tensorrt_path}' is not a valid path.")
if single_model: if single_model:
if faster_whisper_custom_model_path or whisper_tensorrt_path: if faster_whisper_custom_model_path or whisper_tensorrt_path:
logging.info("Custom model option was provided. Switching to single model mode.") logging.info("Custom model option was provided. Switching to single model mode.")
self.single_model = True self.single_model = True
# TODO: load models initially # TODO: load model initially
else: else:
logging.info("Single model mode currently only works with custom models.") logging.info("Single model mode currently only works with custom models.")
with serve( with serve(