fixed bug with trigger words not removing correctly on deselect

Metal079 · Sep 30, 2024 · 822bdee · 822bdee
1 parent b268b5e
commit 822bdee
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/app/home/options/options.component.ts b/src/app/home/options/options.component.ts
@@ -1448,7 +1448,11 @@ export class OptionsComponent implements OnInit {
 
     // Remove the trigger prompt from the prompt input
     if (lora.trigger_words) {
-      lora.trigger_words.forEach((element: String) => {
+      // sanitize trigger words
+      let trigger_words = lora.trigger_words;
+      trigger_words = trigger_words.map((word: string) => word.trim().toLowerCase());
+
+      trigger_words.forEach((element: String) => {
         this.generationRequest.prompt = this.generationRequest.prompt.replace(', ' + element, '');
       });
       this.sharedService.setPrompt(this.generationRequest.prompt);