Spaces:
Sleeping
Sleeping
honeysuckle
commited on
Commit
·
3358ce6
1
Parent(s):
48f5a5d
fixes
Browse files
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 |
-
|
| 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=
|
| 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:")
|