UI: More friendly message when no model is loaded

This commit is contained in:
oobabooga 2025-05-09 07:21:05 -07:00
parent 2bde625d57
commit 8984e95c67
3 changed files with 21 additions and 5 deletions

View file

@ -72,6 +72,20 @@ def natural_keys(text):
return [atoi(c) for c in re.split(r'(\d+)', text)]
def check_model_loaded():
if shared.model_name == 'None' or shared.model is None:
if len(get_available_models()) <= 1:
error_msg = "No model is loaded.\n\nTo get started:\n1) Place a GGUF file in your user_data/models folder\n2) Go to the Model tab and select it"
logger.error(error_msg)
return False, error_msg
else:
error_msg = "No model is loaded. Please select one in the Model tab."
logger.error(error_msg)
return False, error_msg
return True, None
def get_available_models():
# Get all GGUF files
gguf_files = get_available_ggufs()