feat: add loras

app.py

@@ -22,7 +22,6 @@ if access_token:
 else:
     print("warning: no access token found")
 
-
 # Load LoRAs from JSON file
 with open('loras.json', 'r') as f:
     loras = json.load(f)
@@ -59,12 +58,12 @@ class calculateDuration:
 def update_selection(evt: gr.SelectData, width, height):
     selected_lora = loras[evt.index]
 
-    new_placeholder = selected_lora.get('placeholder', f"Type a prompt for {selected_lora['title']}")
-    example_prompt = selected_lora.get('example_prompt', '')
+    # update placeholder
+    new_placeholder = f"Type a prompt for {selected_lora['title']}, trigger word is {selected_lora['trigger_word']}"
     lora_repo = selected_lora["repo"]
     updated_text = f"### Selected: [{lora_repo}](https://huggingface.co/{lora_repo}) ✨"
+    example_prompt = selected_lora.get("example_prompt", "Enter your prompt here...")
 
-    # 分辨率切换
     if "aspect" in selected_lora:
         if selected_lora["aspect"] == "portrait":
             width = 768
@@ -75,12 +74,10 @@ def update_selection(evt: gr.SelectData, width, height):
         else:
             width = 1024
             height = 1024
-            
-    gr.Info("LoRA selection updated")  # 添加这行来触发 UI 刷新
 
     return (
-        gr.update(placeholder=new_placeholder, value=example_prompt),
-        updated_text,
+        gr.update(value=example_prompt, placeholder=new_placeholder),
+        gr.update(value=updated_text),
         evt.index,
         width,
         height,
@@ -105,15 +102,15 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scal
         ):
             yield img
 
-#  执行 run 
 def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
     if selected_index is None:
         raise gr.Error("You must select a LoRA before proceeding.")
     selected_lora = loras[selected_index]
+    gr.Info(f"Current LoRA: {selected_lora['title']}")
 
     lora_path = selected_lora["repo"]
     trigger_word = selected_lora["trigger_word"]
-
+    
     if(trigger_word):
         if "trigger_position" in selected_lora:
             if selected_lora["trigger_position"] == "prepend":
@@ -127,6 +124,8 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
 
     with calculateDuration("Unloading LoRA"):
         pipe.unload_lora_weights()
+        # 清除 CUDA 缓存
+        torch.cuda.empty_cache()  
         
     # Load LoRA weights
     with calculateDuration(f"Loading LoRA weights for {selected_lora['title']}"):
@@ -134,6 +133,8 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
             pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"])
         else:
             pipe.load_lora_weights(lora_path)
+     # 添加短暂延迟确保 LoRA 完全加载
+    time.sleep(0.5)
 
     # Set random seed for reproducibility
     with calculateDuration("Randomizing seed"):
@@ -245,19 +246,18 @@ css = '''
 .progress-container {width: 100%;height: 30px;background-color: #f0f0f0;border-radius: 15px;overflow: hidden;margin-bottom: 20px}
 .progress-bar {height: 100%;background-color: #4f46e5;width: calc(var(--current) / var(--total) * 100%);transition: width 0.5s ease-in-out}
 '''
+
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
     title = gr.HTML(
-        """<h1><img src="https://huggingface.co/spaces/multimodalart/flux-lora-the-explorer/resolve/main/flux_lora.png" alt="LoRA"> FLUX LoRA the Explorer</h1>""",
+        """<h1><img src="https://huggingface.co/spaces/multimodalart/flux-lora-the-explorer/resolve/main/flux_lora.png" alt="LoRA"> FLUX LoRA Collection </h1>""",
         elem_id="title",
     )
     selected_index = gr.State(None)
-
     with gr.Row():
         with gr.Column(scale=3):
-            prompt = gr.Textbox(label="Prompt", lines=1, placeholder="Type a prompt after selecting a LoRA")
+            prompt = gr.Textbox(label="Prompt", lines=3, placeholder="Type a prompt after selecting a LoRA",interactive=True)
         with gr.Column(scale=1, elem_id="gen_column"):
             generate_button = gr.Button("Generate", variant="primary", elem_id="gen_btn")
-
     with gr.Row():
         with gr.Column():
             selected_info = gr.Markdown("")
@@ -269,11 +269,11 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
                 elem_id="gallery"
             )
             with gr.Group():
-                custom_lora = gr.Textbox(label="Custom LoRA", info="LoRA Hugging Face path", placeholder="XLabs-AI/flux-RealismLora")
+                custom_lora = gr.Textbox(label="Custom LoRA", info="LoRA Hugging Face path", placeholder="multimodalart/vintage-ads-flux")
                 gr.Markdown("[Check the list of FLUX LoRas](https://huggingface.co/models?other=base_model:adapter:black-forest-labs/FLUX.1-dev)", elem_id="lora_list")
             custom_lora_info = gr.HTML(visible=False)
             custom_lora_button = gr.Button("Remove custom LoRA", visible=False)
-
+            
             with gr.Accordion("Advanced Settings", open=False):
                 with gr.Row():
                     cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, step=0.5, value=3.5)
@@ -287,7 +287,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
                     randomize_seed = gr.Checkbox(True, label="Randomize seed")
                     seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
                     lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=3, step=0.01, value=0.95)
-
+                    
         with gr.Column():
             progress_bar = gr.Markdown(elem_id="progress",visible=False)
             result = gr.Image(label="Generated Image")

loras.json

@@ -45,10 +45,11 @@
   },
 
   {
-    "image": "https://huggingface.co/Shakker-Labs/AWPortrait-FL/resolve/main/cover.jpeg",
-    "title": "AWPortrait FL",
-    "repo": "Shakker-Labs/AWPortrait-FL",
-    "weights": "AWPortrait-FL-lora.safetensors",
-    "trigger_word": ""
+    "image": "https://cdn-uploads.huggingface.co/production/uploads/6413514705147e9f74df6fec/_ht43YxR9-K_915vk9_K2.png",
+    "title": "JayChow",
+    "repo": "vincenthugging/flux-dev-lora-jaychow",
+    "trigger_word": "jaychow",
+    "example_prompt": "photo of JAYCHOW, a famous male singer of China, street wear, holding a  handwritten sign  with red text that says 'Hello World!',face detailed ",
+    "placeholder": "Trigger word: jaychow"
   }
 ]