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 |
|