vrr/koboldcpp
2
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2025-09-11 09:34:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

allow forced positive prompt

Browse source
This commit is contained in:
Concedo 2024-06-27 20:21:17 +08:00
parent e433afb261
commit 1801594972
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
koboldcpp.py
View file

@ -591,7 +591,8 @@ def sd_generate(genparams):
default_adapter = {} if chatcompl_adapter is None else chatcompl_adapter default_adapter = {} if chatcompl_adapter is None else chatcompl_adapter
adapter_obj = genparams.get('adapter', default_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") prompt = genparams.get("prompt", "high quality")
negative_prompt = genparams.get("negative_prompt", "") negative_prompt = genparams.get("negative_prompt", "")
@ -600,6 +601,11 @@ def sd_generate(genparams):
negative_prompt += ", " + forced_negprompt negative_prompt += ", " + forced_negprompt
else: else:
negative_prompt = forced_negprompt 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_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]) 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) denoising_strength = genparams.get("denoising_strength", 0.6)