From 388acc4ddd8df80c09a7ed54e3716008c6e9abb8 Mon Sep 17 00:00:00 2001
From: Ash <webartboy@gmail.com>
Date: Fri, 17 Feb 2023 19:04:50 +0400
Subject: [PATCH] optimisation checking EOS tag

---
 public/index.html | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/public/index.html b/public/index.html
index 0251e90a1..496e66fa0 100644
--- a/public/index.html
+++ b/public/index.html
@@ -1021,7 +1021,7 @@
                                     if(is_pygmalion){
                                         if_typing_text = false;
                                         message_already_generated +=getMessage;
-                                        if(message_already_generated.indexOf('You:') === -1 && tokens_already_generated < parseInt(amount_gen) && getMessage.length > 0){
+                                        if(message_already_generated.indexOf('You:') === -1 && message_already_generated.indexOf('<|endoftext|>') === -1 && tokens_already_generated < parseInt(amount_gen) && getMessage.length > 0){
                                             runGenerate(getMessage);
                                             return;
                                         }
@@ -1038,6 +1038,10 @@
                                     if(getMessage.indexOf(name1+":") != -1){ 
                                         getMessage = getMessage.substr(0,getMessage.indexOf(name1+":"));
 
+                                    }
+                                    if(getMessage.indexOf('<|endoftext|>') != -1){ 
+                                        getMessage = getMessage.substr(0,getMessage.indexOf('<|endoftext|>'));
+
                                     }
                                     let this_mes_is_name = true;
                                     if(getMessage.indexOf(name2+":") === 0){