Commit
·
1046573
1
Parent(s):
ca790c6
Refactor flux_events.py and image_tab.py to update LoRA gallery functionality
Browse files- modules/events/flux_events.py +9 -0
- tabs/image_tab.py +2 -2
modules/events/flux_events.py
CHANGED
@@ -23,6 +23,15 @@ def update_fast_generation(fast_generation):
|
|
23 |
value=8
|
24 |
)
|
25 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
26 |
|
27 |
|
28 |
def selected_lora_from_gallery(evt: gr.SelectData):
|
|
|
23 |
value=8
|
24 |
)
|
25 |
)
|
26 |
+
else:
|
27 |
+
return (
|
28 |
+
gr.update(
|
29 |
+
value=3.5
|
30 |
+
),
|
31 |
+
gr.update(
|
32 |
+
value=20
|
33 |
+
)
|
34 |
+
)
|
35 |
|
36 |
|
37 |
def selected_lora_from_gallery(evt: gr.SelectData):
|
tabs/image_tab.py
CHANGED
@@ -54,7 +54,7 @@ def flux_tab():
|
|
54 |
with gr.Column(scale=2):
|
55 |
globals()[f"lora_slider_{i}"] = gr.Slider(label=f"LoRA {i+1}", minimum=0, maximum=1, step=0.01, value=0.8, visible=False, interactive=True)
|
56 |
with gr.Column():
|
57 |
-
globals()[f"lora_remove_{i}"] = gr.Button(value="Remove LoRA", visible=False)
|
58 |
|
59 |
|
60 |
with gr.Accordion("Embeddings", open=False): # Embeddings
|
@@ -150,7 +150,7 @@ def flux_tab():
|
|
150 |
|
151 |
# Lora Gallery
|
152 |
lora_gallery.select(selected_lora_from_gallery, None, selected_lora)
|
153 |
-
custom_lora.change(update_selected_lora, custom_lora, [selected_lora,
|
154 |
add_lora.click(add_to_enabled_loras, [selected_lora, enabled_loras], [selected_lora, custom_lora_info, enabled_loras])
|
155 |
enabled_loras.change(update_lora_sliders, enabled_loras, [lora_slider_0, lora_slider_1, lora_slider_2, lora_slider_3, lora_slider_4, lora_slider_5, lora_remove_0, lora_remove_1, lora_remove_2, lora_remove_3, lora_remove_4, lora_remove_5]) # type: ignore
|
156 |
|
|
|
54 |
with gr.Column(scale=2):
|
55 |
globals()[f"lora_slider_{i}"] = gr.Slider(label=f"LoRA {i+1}", minimum=0, maximum=1, step=0.01, value=0.8, visible=False, interactive=True)
|
56 |
with gr.Column():
|
57 |
+
globals()[f"lora_remove_{i}"] = gr.Button(value="Remove LoRA", visible=False)
|
58 |
|
59 |
|
60 |
with gr.Accordion("Embeddings", open=False): # Embeddings
|
|
|
150 |
|
151 |
# Lora Gallery
|
152 |
lora_gallery.select(selected_lora_from_gallery, None, selected_lora)
|
153 |
+
custom_lora.change(update_selected_lora, custom_lora, [selected_lora, custom_lora_info])
|
154 |
add_lora.click(add_to_enabled_loras, [selected_lora, enabled_loras], [selected_lora, custom_lora_info, enabled_loras])
|
155 |
enabled_loras.change(update_lora_sliders, enabled_loras, [lora_slider_0, lora_slider_1, lora_slider_2, lora_slider_3, lora_slider_4, lora_slider_5, lora_remove_0, lora_remove_1, lora_remove_2, lora_remove_3, lora_remove_4, lora_remove_5]) # type: ignore
|
156 |
|