From 1801594972db823d00fdfa4c65eb3f541151749c Mon Sep 17 00:00:00 2001 From: Concedo <39025047+LostRuins@users.noreply.github.com> Date: Thu, 27 Jun 2024 20:21:17 +0800 Subject: [PATCH] allow forced positive prompt --- koboldcpp.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/koboldcpp.py b/koboldcpp.py index 4afe533b1..8af2a941a 100644 --- a/koboldcpp.py +++ b/koboldcpp.py @@ -591,7 +591,8 @@ def sd_generate(genparams): default_adapter = {} if chatcompl_adapter is None else chatcompl_adapter adapter_obj = genparams.get('adapter', default_adapter) - forced_negprompt = adapter_obj.get("negative_prompt", "") + forced_negprompt = adapter_obj.get("add_sd_negative_prompt", "") + forced_posprompt = adapter_obj.get("add_sd_prompt", "") prompt = genparams.get("prompt", "high quality") negative_prompt = genparams.get("negative_prompt", "") @@ -600,6 +601,11 @@ def sd_generate(genparams): negative_prompt += ", " + forced_negprompt else: negative_prompt = forced_negprompt + if forced_posprompt!="": + if prompt!="": + prompt += ", " + forced_posprompt + else: + prompt = forced_posprompt init_images_arr = genparams.get("init_images", []) init_images = ("" if (not init_images_arr or len(init_images_arr)==0 or not init_images_arr[0]) else init_images_arr[0]) denoising_strength = genparams.get("denoising_strength", 0.6)