From a069f3904cb0d744055b39319b30fdea091ce91d Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sun, 17 Sep 2023 08:12:08 -0700
Subject: [PATCH] Undo part of ad8ac545a58fde3a26bbe3a04489131fe1b3ec88

---
 modules/text_generation.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/modules/text_generation.py b/modules/text_generation.py
index 5296c1b4..d0f37007 100644
--- a/modules/text_generation.py
+++ b/modules/text_generation.py
@@ -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