Enable single-model mode for stock models when batch_inference is set
Signed-off-by: makaveli10 <vineet.suryan@collabora.com>
This commit is contained in:
@@ -663,6 +663,9 @@ class TranscriptionServer:
|
|||||||
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 model initially
|
# TODO: load model initially
|
||||||
|
elif batch_enabled:
|
||||||
|
logging.info("Batch inference enabled. Switching to single model mode for stock model.")
|
||||||
|
self.single_model = True
|
||||||
else:
|
else:
|
||||||
logging.info("Single model mode currently only works with custom models.")
|
logging.info("Single model mode currently only works with custom models.")
|
||||||
if not BackendType.is_valid(backend):
|
if not BackendType.is_valid(backend):
|
||||||
|
|||||||
Reference in New Issue
Block a user