adding trigger token config param

app.py

@@ -29,6 +29,7 @@ def models_change(model, scheduler, config):
 
     use_safetensors = False
     refiner = "none"
+    trigger_token = ""
     
     # no model selected (because this is UI init run)
     if type(model) != list and str(model) != 'None':
@@ -36,7 +37,8 @@ def models_change(model, scheduler, config):
         use_safetensors = str(models[model]['use_safetensors'])
         model_description = models[model]['description']
         refiner = models[model]['refiner']
-                
+        trigger_token = models[model]['trigger_token'] 
+        
         # if no scheduler is selected, choose the default one for this model
         if scheduler == None:
             
@@ -53,7 +55,7 @@ def models_change(model, scheduler, config):
     # safety_checker_change(in_safety_checker.value, config)
     # requires_safety_checker_change(in_requires_safety_checker.value, config)
 
-    return model_description, refiner, use_safetensors, scheduler, config, str(config), assemble_code(config)
+    return model_description, refiner, trigger_token, use_safetensors, scheduler, config, str(config), assemble_code(config)
 
 def data_type_change(data_type, config):
 
@@ -91,6 +93,12 @@ def prompt_change(prompt, config):
 
     return config, str(config), assemble_code(config)
 
+def trigger_token_change(trigger_token, config):
+    
+    config = set_config(config, 'trigger_token', trigger_token)
+
+    return config, str(config), assemble_code(config)
+
 def negative_prompt_change(negative_prompt, config):
     
     config = set_config(config, 'negative_prompt', negative_prompt)
@@ -226,10 +234,11 @@ with gr.Blocks(analytics_enabled=False) as demo:
         out_model_description = gr.Textbox(value="", label="Description")
     with gr.Row():
         with gr.Column(scale=1):
+            in_trigger_token = gr.Textbox(value=config.value["trigger_token"], label="Trigger Token")
             in_use_safetensors = gr.Radio(label="Use safe tensors:", choices=["True", "False"], interactive=False)
-            in_cpu_offload = gr.Radio(label="CPU Offload:", value=config.value["cpu_offload"], choices=["True", "False"], info="This may increase performance, as it offloads computations from the GPU to the CPU. But this can also lead to slower executions and lower effectiveness. Compare running time and outputs before making sure, that this setting will help you")
             in_model_refiner = gr.Dropdown(value=config.value["refiner"], choices=["none"], label="Refiner", allow_custom_value=True, multiselect=False)
         with gr.Column(scale=1):
+            in_cpu_offload = gr.Radio(label="CPU Offload:", value=config.value["cpu_offload"], choices=["True", "False"], info="This may increase performance, as it offloads computations from the GPU to the CPU. But this can also lead to slower executions and lower effectiveness. Compare running time and outputs before making sure, that this setting will help you")
             in_safety_checker = gr.Radio(label="Enable safety checker:", value=config.value["safety_checker"], choices=["True", "False"])
             in_requires_safety_checker = gr.Radio(label="Requires safety checker:", value=config.value["requires_safety_checker"], choices=["True", "False"])
 
@@ -268,7 +277,7 @@ with gr.Blocks(analytics_enabled=False) as demo:
     in_data_type.change(data_type_change, inputs=[in_data_type, config], outputs=[config, out_config, out_code])
     in_allow_tensorfloat32.change(tensorfloat32_change, inputs=[in_allow_tensorfloat32, config], outputs=[config, out_config, out_code])
     in_variant.change(variant_change, inputs=[in_variant, config], outputs=[config, out_config, out_code])
-    in_models.change(models_change, inputs=[in_models, in_schedulers, config], outputs=[out_model_description, in_model_refiner, in_use_safetensors, in_schedulers, config, out_config, out_code])
+    in_models.change(models_change, inputs=[in_models, in_schedulers, config], outputs=[out_model_description, in_model_refiner, in_trigger_token, in_use_safetensors, in_schedulers, config, out_config, out_code])
     in_model_refiner.change(model_refiner_change, inputs=[in_model_refiner, config], outputs=[config, out_config, out_code])
     in_cpu_offload.change(cpu_offload_change, inputs=[in_cpu_offload, config], outputs=[config, out_config, out_code])
     in_safety_checker.change(safety_checker_change, inputs=[in_safety_checker, config], outputs=[config, out_config, out_code])
@@ -278,6 +287,7 @@ with gr.Blocks(analytics_enabled=False) as demo:
     in_manual_seed.change(manual_seed_change, inputs=[in_manual_seed, config], outputs=[config, out_config, out_code])
     in_guidance_scale.change(guidance_scale_change, inputs=[in_guidance_scale, config], outputs=[config, out_config, out_code])
     in_prompt.change(prompt_change, inputs=[in_prompt, config], outputs=[config, out_config, out_code])
+    in_trigger_token.change(trigger_token_change, inputs=[in_trigger_token, config], outputs=[config, out_config, out_code])
     in_negative_prompt.change(negative_prompt_change, inputs=[in_negative_prompt, config], outputs=[config, out_config, out_code])
     ev_run_inference = btn_start_pipeline.click(run_inference, inputs=[config, config_history], outputs=[out_image, out_config_history, config_history])
     btn_stop_pipeline.click(fn=None, inputs=None, outputs=None, cancels=[ev_run_inference])
@@ -298,6 +308,7 @@ with gr.Blocks(analytics_enabled=False) as demo:
             in_requires_safety_checker,
             in_schedulers,
             in_prompt,
+            in_trigger_token,
             in_negative_prompt,
             in_inference_steps,
             in_manual_seed,

appConfig.json

@@ -5,55 +5,64 @@
             "use_safetensors": false,
             "description": "see https://huggingface.co/sd-dreambooth-library/solo-levelling-art-style",
             "scheduler": "DDIMScheduler",
-            "refiner": "none"
+            "refiner": "none",
+            "trigger_token": ""
         },
         "CompVis/stable-diffusion-v1-4": {
             "use_safetensors": true,
             "description": "see https://huggingface.co/CompVis/stable-diffusion-v1-4",
             "scheduler": "EulerDiscreteScheduler",
-            "refiner": "none"
+            "refiner": "none",
+            "trigger_token": ""
         },
         "runwayml/stable-diffusion-v1-5": {
             "use_safetensors": true,
             "description": "see https://huggingface.co/runwayml/stable-diffusion-v1-5",
             "scheduler": "DDPMScheduler",
-            "refiner": "none"
+            "refiner": "none",
+            "trigger_token": ""
         },
         "stabilityai/stable-diffusion-2-1": {
             "use_safetensors": true,
             "description": "see https://huggingface.co/stabilityai/stable-diffusion-2-1",
             "scheduler": "DPMSolverMultistepScheduler",
-            "refiner": "none"
+            "refiner": "none",
+            "trigger_token": ""
         },
         "stabilityai/stable-diffusion-xl-base-1.0": {
             "use_safetensors": true,
             "description": "see https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0",
             "scheduler": "DDPMScheduler",
-            "refiner": "stabilityai/stable-diffusion-xl-refiner-1.0"
+            "refiner": "stabilityai/stable-diffusion-xl-refiner-1.0",
+            "trigger_token": ""
         },
         "sd-dreambooth-library/house-emblem": {
             "use_safetensors": false,
             "description": "see https://huggingface.co/sd-dreambooth-library/house-emblem",
             "scheduler": "DDPMScheduler",
-            "refiner": "none"
+            "refiner": "none",
+            "trigger_token": ""
         },
         "Envvi/Inkpunk-Diffusion": {
             "use_safetensors": false,
             "description": "see https://huggingface.co/Envvi/Inkpunk-Diffusion",
             "scheduler": "DDPMScheduler",
-            "refiner": "none"
+            "refiner": "none",
+            "trigger_token": ""
         },
         "Stelath/textual_inversion_comic_strip_fp16": {
             "use_safetensors": true,
             "description": "see https://huggingface.co/Stelath/textual_inversion_comic_strip_fp16",
             "scheduler": "DDPMScheduler",
-            "refiner": "none"
+            "refiner": "none",
+            "trigger_token": ""
         },
         "sd-dreambooth-library/herge-style": {
             "use_safetensors": false,
             "description": "see https://huggingface.co/sd-dreambooth-library/herge-style",
             "scheduler": "DDPMScheduler",
-            "refiner": "none"
+            "refiner": "none",
+            "trigger_token": "herge_style"
         }
 
     },

config.py

@@ -50,6 +50,7 @@ def get_initial_config():
         "inference_steps": 10,
         "guidance_scale": 0.5,
         "prompt": 'A white rabbit',
+        "trigger_token": '',
         "negative_prompt": 'lowres, cropped, worst quality, low quality, chat bubble, chat bubbles, ugly',
     }
     
@@ -87,6 +88,7 @@ def get_config_from_url(initial_config, request: Request):
             return_config['requires_safety_checker'],
             return_config['scheduler'],
             return_config['prompt'],
+            return_config['trigger_token'],
             return_config['negative_prompt'],
             return_config['inference_steps'],
             return_config['manual_seed'],
@@ -173,7 +175,7 @@ def assemble_code(str_config):
         code.append(f'manual_seed = {config["manual_seed"]}')
         code.append(f'generator = torch.manual_seed(manual_seed)')
 
-    code.append(f'prompt = "{config["prompt"]}"')
+    code.append(f'prompt = "{config["prompt"]} {config["trigger_token"]}"')
     code.append(f'negative_prompt = "{config["negative_prompt"]}"')
     code.append(f'inference_steps = {config["inference_steps"]}')
     code.append(f'guidance_scale = {config["guidance_scale"]}')