mantrakp commited on
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 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, custom_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