WIP: Prevent overflooding max_length and throwing error.

chat.py

@@ -172,6 +172,11 @@ def _generate(
             state["history"] = orig_history + [[prompt, new_line + stats()]]
             yield state, state["history"], "", output
 
+            # Avoid throwing an exception by generate()
+            # to prevent UI errors.
+            if cnt >= max_length - 6: # FIXME Bulgarian constant
+                break
+
         # Final line w/o statistics
         yield state, state["history"], "", output
 

prompt.py

@@ -109,6 +109,11 @@ def _generate(
 
             yield state, prompt2, output + stats()
 
+            # Avoid throwing exception by generate()
+            # to prevent UI errors.
+            if cnt >= max_length - 6: # FIXME bulgarian constant
+                break
+
         # Prints final result w/o statistics
         yield state, prompt2, output
     except Exception as e: