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

add negative prompt support to chat completions adapter

Browse source
This commit is contained in:
Concedo 2024-06-26 11:12:24 +08:00
parent 6cf917bbf0
commit e42bc5d677
1 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
koboldcpp.py
View file

@ -587,9 +587,19 @@ def sd_load_model(model_filename,vae_filename,lora_filename):
return ret return ret
def sd_generate(genparams): def sd_generate(genparams):
global maxctx, args, currentusergenkey, totalgens, pendingabortkey global maxctx, args, currentusergenkey, totalgens, pendingabortkey, chatcompl_adapter
default_adapter = {} if chatcompl_adapter is None else chatcompl_adapter
adapter_obj = genparams.get('adapter', default_adapter)
forced_negprompt = adapter_obj.get("negative_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", "")
if forced_negprompt!="":
if negative_prompt!="":
negative_prompt += " " + forced_negprompt
else:
negative_prompt = forced_negprompt
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)