mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2026-04-05 14:45:28 +00:00
UI: More friendly message when no model is loaded
This commit is contained in:
parent
2bde625d57
commit
8984e95c67
3 changed files with 21 additions and 5 deletions
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue