simcop2387/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity

Undo part of ad8ac545a5

Browse source
This commit is contained in:
oobabooga 2023-09-17 08:12:08 -07:00
parent e34c6e6938
commit a069f3904c
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
modules/text_generation.py
View file

@ -39,7 +39,8 @@ def _generate_reply(question, state, stopping_strings=None, is_chat=False, escap
if generate_func is None:
if shared.model_name == 'None' or shared.model is None:
logger.error("No model is loaded! Select one in the Model tab.")
raise ValueError('No model is loaded! Select one in the Model tab.')
yield ''
return
if shared.model.__class__.__name__ in ['LlamaCppModel', 'RWKVModel', 'ExllamaModel', 'Exllamav2Model', 'CtransformersModel']:
generate_func = generate_reply_custom