Upload 4 files

app.py

@@ -104,6 +104,9 @@ def update_selection(evt: gr.SelectData, width, height):
         elif selected_lora["aspect"] == "landscape":
             width = 1024
             height = 768
+        else:
+            width = 1024
+            height = 1024
     return (
         gr.update(placeholder=new_placeholder),
         updated_text,
@@ -324,7 +327,7 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', fill_width=True, css=css) as app:
             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(scale=3):
+            with gr.Column():
                 selected_info = gr.Markdown("")
                 gallery = gr.Gallery(
                     [(item["image"], item["title"]) for item in loras],
@@ -339,7 +342,7 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', fill_width=True, css=css) as app:
                 custom_lora_info = gr.HTML(visible=False)
                 custom_lora_button = gr.Button("Remove custom LoRA", visible=False)
                 deselect_lora_button = gr.Button("Deselect LoRA", variant="secondary")
-            with gr.Column(scale=4):
+            with gr.Column():
                 result = gr.Image(label="Generated Image", format="png", show_share_button=False)
         with gr.Group():
             model_name = gr.Dropdown(label="Base Model", info="You can enter a huggingface model repo_id to want to use.", choices=models, value=models[0], allow_custom_value=True)
@@ -395,7 +398,7 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', fill_width=True, css=css) as app:
                                     lora_download = [None] * num_loras
                                     for i in range(num_loras):
                                         lora_download[i] = gr.Button(f"Get and set LoRA to {int(i+1)}")
-                    with gr.Accordion("ControlNet (🚧Under construction...🚧)", open=False):
+                    with gr.Accordion("ControlNet (🚧Under construction...🚧)", open=False, visible=False):
                         with gr.Column():
                             cn_on = gr.Checkbox(False, label="Use ControlNet")
                             cn_mode = [None] * num_cns

loras.json

@@ -27,6 +27,13 @@
     "weights": "ovld_style_overlord_wavymulder.safetensors",
     "trigger_word": ", ovld style anime"
   },
+  {
+    "image": "https://huggingface.co/Norod78/JojosoStyle-flux-lora/resolve/main/samples/1725244218477__000004255_1.jpg",
+    "title": "JoJo's Bizarre Adventure: Stone Ocean ",
+    "repo": "Norod78/JojosoStyle-flux-lora",
+    "weights": "JojosoStyle_flux_lora.safetensors",
+    "trigger_word": ", JojosoStyle"
+  },
   {
     "image": "https://huggingface.co/multimodalart/flux-tarot-v1/resolve/main/images/e5f2761e5d474e6ba492d20dca0fa26f_e78f1524074b42b6ac49643ffad50ac6.png",
     "title": "Tarot v1",
@@ -47,19 +54,6 @@
     "trigger_word": "RU883R style",
     "trigger_position": "prepend"
   },
-  {
-    "image": "https://huggingface.co/mgwr/Cine-Aesthetic/resolve/main/images/00030-1333633802.png",
-    "title": "Cine Aesthetic",
-    "repo": "mgwr/Cine-Aesthetic",
-    "trigger_word": "mgwr/cine",
-    "trigger_position": "prepend"
-  },
-  {
-    "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-blended-realistic-illustration/resolve/main/images/example3.png",
-    "title": "Blended Realistic Illustration",
-    "repo": "Shakker-Labs/FLUX.1-dev-LoRA-blended-realistic-illustration",
-    "trigger_word": "artistic style blends reality and illustration elements"
-  },
   {
     "image": "https://github.com/XLabs-AI/x-flux/blob/main/assets/readme/examples/picture-6-rev1.png?raw=true",
     "title": "flux-Realism",
@@ -73,6 +67,13 @@
     "trigger_word": "a vintage ad of",
     "trigger_position": "prepend"
   },
+  {
+    "image": "https://huggingface.co/mgwr/Cine-Aesthetic/resolve/main/images/00030-1333633802.png",
+    "title": "Cine Aesthetic",
+    "repo": "mgwr/Cine-Aesthetic",
+    "trigger_word": "mgwr/cine",
+    "trigger_position": "prepend"
+  },
   {
     "image": "https://huggingface.co/nerijs/animation2k-flux/resolve/main/images/Q8-oVxNnXvZ9HNrgbNpGw_02762aaaba3b47859ee5fe9403a371e3.png",
     "title": "animation2k",
@@ -89,7 +90,14 @@
     "image": "https://huggingface.co/veryVANYA/ps1-style-flux/resolve/main/24439220.jpeg",
     "title": "PS1 style",
     "repo": "veryVANYA/ps1-style-flux",
-    "trigger_word": "ps1 game screenshot"
+    "trigger_word": "ps1 game screenshot,",
+    "trigger_position": "prepend"
+  },
+  {
+    "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-blended-realistic-illustration/resolve/main/images/example3.png",
+    "title": "Blended Realistic Illustration",
+    "repo": "Shakker-Labs/FLUX.1-dev-LoRA-blended-realistic-illustration",
+    "trigger_word": "artistic style blends reality and illustration elements"
   },
   {
     "image": "https://huggingface.co/alvdansen/flux-koda/resolve/main/images/ComfyUI_00566_%20(2).png",

mod.py

@@ -29,7 +29,7 @@ models = [
     "John6666/niji56-style-v3-fp8-flux",
     "John6666/lyh-dalle-anime-v12dalle-fp8-flux",
     "John6666/glimmerkin-flux-cute-v10-fp8-flux",
-    "John6666/xe-anime-flux-02-fp8-flux",
+    "John6666/xe-anime-flux-03-fp8-flux",
     "John6666/xe-figure-flux-01-fp8-flux",
     "John6666/xe-pixel-flux-01-fp8-flux",
     "John6666/fluxunchained-artfulnsfw-fut516xfp8e4m3fnv11-fp8-flux",
@@ -43,6 +43,7 @@ models = [
     "John6666/jib-mix-flux-v208stephyper-fp8-flux",
     "John6666/flux-asian-realistic-v10-fp8-flux",
     "John6666/fluxasiandoll-v10-fp8-flux",
+    "John6666/xe-asian-flux-01-fp8-flux",
     "John6666/fluxescore-dev-v10fp16-fp8-flux",
     # "",
 ]
@@ -73,8 +74,13 @@ def is_repo_exists(repo_id):
 
 
 def clear_cache():
-    torch.cuda.empty_cache()
-    gc.collect()
+    try:
+        torch.cuda.empty_cache()
+        #torch.cuda.reset_max_memory_allocated()
+        torch.cuda.reset_peak_memory_stats()
+        gc.collect()
+    except Exception as e:
+        print(e)
 
 
 def deselect_lora():
@@ -345,3 +351,4 @@ load_prompt_enhancer.zerogpu = True
 fuse_loras.zerogpu = True
 preprocess_image.zerogpu = True
 get_control_params.zerogpu = True
+clear_cache.zerogpu = True

requirements.txt

@@ -1,6 +1,6 @@
 spaces
 git+https://github.com/huggingface/diffusers
-torch
+torch==2.2.0
 torchvision
 huggingface_hub
 accelerate
@@ -9,9 +9,10 @@ peft
 sentencepiece
 timm
 einops
-controlnet-aux
+controlnet_aux
 kornia
 numpy
 opencv-python
 deepspeed
+mediapipe
 openai==1.37.0