honeysuckle commited on
Commit
3358ce6
·
1 Parent(s): 48f5a5d
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -40,10 +40,8 @@ def prepare_characters(hf_path="OnMoon/loras"):
40
 
41
  return character_names, character_configs
42
 
43
- character_names, character_configs = prepare_characters()
44
 
45
- # Invoke endpoint
46
- request_to_endpoint({"input": {"prompt": ""}})
47
 
48
 
49
  def process_input (name, scale, triggers, tech_prompt, tech_negative_prompt, prompts_count, *args):
@@ -58,6 +56,7 @@ def process_input (name, scale, triggers, tech_prompt, tech_negative_prompt, pro
58
  model = character_configs[name]["model"]
59
  model['loras'] = {name: scale}
60
  params = character_configs[name]["params"]
 
61
  for n in range(prompts_count):
62
  prompts.append(f'{triggers}, {tech_prompt}, {boxes[2*n]}')
63
  negative_prompts.append(f"{tech_negative_prompt}, {boxes[2*n + 1]}")
@@ -108,7 +107,7 @@ with gr.Blocks() as demo:
108
  )
109
 
110
  with gr.Accordion(open=False):
111
- scale = gr.Slider(0, 2.0, 0.01, label=f"{name} scale:", value=0.75, interactive=True)
112
  triggers = gr.Textbox(label=f"Trigger words:")
113
  tech_prompt = gr.Textbox(label=f"Technical prompt:")
114
  tech_negative_prompt = gr.Textbox(label=f"Negative technical prompt:")
 
40
 
41
  return character_names, character_configs
42
 
 
43
 
44
+ character_names, character_configs = prepare_characters()
 
45
 
46
 
47
  def process_input (name, scale, triggers, tech_prompt, tech_negative_prompt, prompts_count, *args):
 
56
  model = character_configs[name]["model"]
57
  model['loras'] = {name: scale}
58
  params = character_configs[name]["params"]
59
+ params["cross_attention_kwargs"]['scale'] = scale
60
  for n in range(prompts_count):
61
  prompts.append(f'{triggers}, {tech_prompt}, {boxes[2*n]}')
62
  negative_prompts.append(f"{tech_negative_prompt}, {boxes[2*n + 1]}")
 
107
  )
108
 
109
  with gr.Accordion(open=False):
110
+ scale = gr.Slider(0, 2.0, 0.75, step=0.01, label="Selected LoRA scale:", interactive=True)
111
  triggers = gr.Textbox(label=f"Trigger words:")
112
  tech_prompt = gr.Textbox(label=f"Technical prompt:")
113
  tech_negative_prompt = gr.Textbox(label=f"Negative technical prompt:")