Commit 1ca724e8 authored by Your Name's avatar Your Name

Fix more model_name vs model_names bugs

Fix remaining occurrences of model_name (singular) being used instead
of model_names (list) in main function.
parent e0530b65
...@@ -4223,7 +4223,7 @@ def main(): ...@@ -4223,7 +4223,7 @@ def main():
# Pre-load first audio model at startup if: # Pre-load first audio model at startup if:
# - Using loadall or loadswap mode, OR # - Using loadall or loadswap mode, OR
# - No main model is specified (only audio model configured) # - No main model is specified (only audio model configured)
should_preload = load_mode in ("loadall", "loadswap") or (model_name is None and audio_models) should_preload = load_mode in ("loadall", "loadswap") or (not model_names and audio_models)
if should_preload: if should_preload:
print(f"Pre-loading audio model... {audio_models[0]}") print(f"Pre-loading audio model... {audio_models[0]}")
...@@ -4391,7 +4391,7 @@ def main(): ...@@ -4391,7 +4391,7 @@ def main():
multi_model_manager.set_tts_model(args.tts_model, {}) multi_model_manager.set_tts_model(args.tts_model, {})
# Pre-load TTS model if it's the only model configured # Pre-load TTS model if it's the only model configured
if model_name is None and not audio_models and not image_models: if not model_names and not audio_models and not image_models:
print(f"Pre-loading TTS model...") print(f"Pre-loading TTS model...")
# TTS models load on-demand, but we can pre-download if needed # TTS models load on-demand, but we can pre-download if needed
...@@ -4410,7 +4410,7 @@ def main(): ...@@ -4410,7 +4410,7 @@ def main():
}) })
# Pre-load image model if it's the only model configured # Pre-load image model if it's the only model configured
if model_name is None and not audio_models and not args.tts_model: if not model_names and not audio_models and not args.tts_model:
print(f"Pre-loading image model...") print(f"Pre-loading image model...")
# Register model aliases if specified # Register model aliases if specified
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment