From d9e898afe0b45ca7d0a25aebeea3f896d63061cb Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Sun, 19 Oct 2025 19:53:40 +0800
Subject: [PATCH] reset scheduler if default otherwise it will persist the old
 one

---
 embd_res/klite.embd                  | 56 +++++++++++++++-------------
 otherarch/sdcpp/stable-diffusion.cpp |  4 +-
 2 files changed, 33 insertions(+), 27 deletions(-)

diff --git a/embd_res/klite.embd b/embd_res/klite.embd
index 58bf2e183..7d9c5cec7 100644
--- a/embd_res/klite.embd
+++ b/embd_res/klite.embd
@@ -15371,23 +15371,26 @@ Current version indicated by LITEVER below.
 		return inputtxt;
 	}
 
-	function end_trim_to_sentence(input,include_newline=false) {
+	function end_trim_to_sentence(input, include_newline = false) {
 		let last = -1;
 		let enders = ['.', '!', '?', '*', '"', ')', '}', '`', ']', ';', '…'];
-		for (let i = 0; i < enders.length; ++i)
-		{
+		for (let i = 0; i < enders.length; ++i) {
 			last = Math.max(last, input.lastIndexOf(enders[i]));
 		}
-
-		if(include_newline)
-		{
+		if (include_newline) {
 			let nl = input.lastIndexOf("\n");
 			last = Math.max(last, nl);
 		}
-		if (last > 0) {
-			return input.substring(0, last + 1).replace(/[\t\r\n ]+$/, '');
+		let cleaned = input.replace(/[\t\r\n ]+$/, '');
+		if (last <= 0) {  // If no valid end found
+			return cleaned;
 		}
-		return input.replace(/[\t\r\n ]+$/, '');
+		let trimmedLength = cleaned.length - (last + 1); // Calculate how many characters would be trimmed
+		if (trimmedLength > 150) {	// If more than 150 chars would be trimmed, skip trimming
+			return cleaned;
+		}
+		// Otherwise, trim up to the last sentence end
+		return cleaned.substring(0, last + 1).replace(/[\t\r\n ]+$/, '');
 	}
 
 	function start_trim_to_sentence(input) {
@@ -18038,29 +18041,30 @@ Current version indicated by LITEVER below.
 							//it's an object/array, so repack the text turn
 							let t1 = mainoaibody[0].text;
 							mhistory = repack_instruct_history(t1);
+						}
 
-							//many backends REFUSE to allow the assistant to go first. If the first turn belongs to assistant, we turn it into user
-							if(mhistory.length>0 && !mhistory[0].myturn)
-							{
-								mhistory[0].myturn = true;
-							}
-							//if both first and second turns are from user, merge them into one turn
-							if(mhistory.length>1 && mhistory[0].myturn && mhistory[1].myturn)
-							{
-								mhistory[0].msg += "\n"+mhistory[1].msg;
-								mhistory.splice(1, 1);
-							}
+						//many backends REFUSE to allow the assistant to go first. If the first turn belongs to assistant, we turn it into user
+						if(mhistory.length>0 && !mhistory[0].myturn)
+						{
+							mhistory[0].myturn = true;
+						}
+						//if both first and second turns are from user, merge them into one turn
+						if(mhistory.length>1 && mhistory[0].myturn && mhistory[1].myturn)
+						{
+							mhistory[0].msg += "\n"+mhistory[1].msg;
+							mhistory.splice(1, 1);
+						}
 
-							if(mhistory.length>0 && mainoaibody.length>1 && mainoaibody[1].type && mainoaibody[1].type!="text")
+						if(mhistory.length>0 && mainoaibody.length>1 && mainoaibody[1].type && mainoaibody[1].type!="text")
+						{
+							mhistory[0].msg = [{type: 'text', text: mhistory[0].msg}];
+							for(let i=1;i<mainoaibody.length;++i)
 							{
-								mhistory[0].msg = [{type: 'text', text: mhistory[0].msg}];
-								for(let i=1;i<mainoaibody.length;++i)
-								{
-									mhistory[0].msg.push(mainoaibody[i]);
-								}
+								mhistory[0].msg.push(mainoaibody[i]);
 							}
 						}
 
+
 						for(let i=0;i<mhistory.length;++i)
 						{
 							oai_payload.messages.push({ "role": (mhistory[i].myturn?"user":"assistant"), "content": mhistory[i].msg });
diff --git a/otherarch/sdcpp/stable-diffusion.cpp b/otherarch/sdcpp/stable-diffusion.cpp
index c20a0c146..db788c99b 100644
--- a/otherarch/sdcpp/stable-diffusion.cpp
+++ b/otherarch/sdcpp/stable-diffusion.cpp
@@ -948,7 +948,9 @@ public:
                 denoiser->scheduler = std::make_shared<SmoothStepSchedule>();
                 break;
             case DEFAULT:
-                // Don't touch anything.
+                // Reset back to discrete
+                LOG_INFO("running with discrete scheduler");
+                denoiser->scheduler = std::make_shared<DiscreteSchedule>();
                 break;
             default:
                 LOG_ERROR("Unknown scheduler %i", scheduler);