Updated image generation functions with extra configurable settings for:

koboldai_vars.img_gen_art_guide
koboldai_vars.img_gen_negative_prompt (local SD API only until Horde supports passing negative prompts.)
koboldai_vars.img_gen_steps
koboldai_vars.img_gen_cfg_scale
Added toggle for the image_generating flag to allow resetting it in the case of interruption without needing to restart KoboldAI.
- Found error in payload format for SD api post request and resolved by consolidating prompt and settings values into single object.
- Refactored new variables to be accessed directly in functions instead of copying to local vars.
- New variables are now working correctly with SD api requests. Need testing with horde and local.

Signed-off-by: viningr
This commit is contained in:
Robert Vining
2022-11-15 16:44:21 +10:00
parent 7ab86a80ff
commit edf04bfb06
3 changed files with 98 additions and 14 deletions

View File

@@ -993,7 +993,11 @@ class user_settings(settings):
self.beep_on_complete = False
self.img_gen_priority = 1
self.show_budget = False
self.img_gen_api_url = "http://127.0.0.1:7860/"
self.img_gen_api_url = "http://127.0.0.1:7860"
self.img_gen_art_guide = "fantasy illustration, artstation, by jason felix by steve argyle by tyler jacobson by peter mohrbacher, cinematic lighting"
self.img_gen_negative_prompt = "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, artist name"
self.img_gen_steps = 30
self.img_gen_cfg_scale = 7
self.cluster_requested_models = [] # The models which we allow to generate during cluster mode