This commit is contained in:
Cyberes 2023-10-05 18:05:31 -06:00
parent a229b4d6c5
commit 01fb619b9b
1 changed files with 4 additions and 0 deletions

View File

@ -99,6 +99,10 @@ def home():
base_client_api = redis.get('base_client_api', dtype=str) base_client_api = redis.get('base_client_api', dtype=str)
stats = generate_stats() stats = generate_stats()
model_choices, default_model = get_model_choices() model_choices, default_model = get_model_choices()
if not model_choices.get(default_model):
return 'The server is still starting up. Please wait...'
default_model_info = model_choices[default_model] default_model_info = model_choices[default_model]
if default_model_info['queued'] == 0 and default_model_info['queued'] >= default_model_info['concurrent_gens']: if default_model_info['queued'] == 0 and default_model_info['queued'] >= default_model_info['concurrent_gens']: