Upload folder using huggingface_hub

README.md

@@ -1,5 +1,5 @@
 ---
-title: ImageGen - FLUX.2-KV
+title: ImageGen
 emoji: 🖼
 colorFrom: purple
 colorTo: red
@@ -7,4 +7,5 @@ sdk: gradio
 sdk_version: "5.50.0"
 app_file: app.py
 short_description: Multi-task image generator with dynamic, chainable workflows
+pinned: true
 ---

ui/layout.py

@@ -10,14 +10,9 @@ def build_ui(event_handler_function):
     ui_components = {}
 
     with gr.Blocks() as demo:
-        gr.Markdown("# ImageGen - FLUX.2-KV")
+        gr.Markdown("# ImageGen")
         gr.Markdown(
-            "This demo is a streamlined version of the [Comfy web UI](https://github.com/RioShiina47/comfy-webui)'s [ImageGen](https://huggingface.co/spaces/RioShiina/ImageGen) functionality. "
-            "Other versions are also available: "
-            "[Anima](https://huggingface.co/spaces/RioShiina/ImageGen-Anima), "
-            "[Illustrious](https://huggingface.co/spaces/RioShiina/ImageGen-Illustrious), "
-            "[NoobAI](https://huggingface.co/spaces/RioShiina/ImageGen-NoobAI), "
-            "[Pony](https://huggingface.co/spaces/RioShiina/ImageGen-Pony)"
+            "This demo is a streamlined version of the [Comfy web UI](https://github.com/RioShiina47/comfy-webui)'s ImageGen functionality. "
         )
         with gr.Tabs(elem_id="tabs_container") as tabs:
             with gr.TabItem("Txt2Img", id=0):

ui/shared/ui_components.py

@@ -50,8 +50,7 @@ def create_model_architecture_filter_ui(prefix):
         label="Model Architecture",
         choices=choices,
         value="ALL",
-        interactive=True,
-        visible=False
+        interactive=True
     )
     return components
 
@@ -150,6 +149,7 @@ def create_controlnet_ui(prefix: str, max_units=MAX_CONTROLNETS):
 
     with gr.Accordion("ControlNet Settings", open=False, visible=('controlnet' in default_enabled_chains)) as accordion:
         components[key('controlnet_accordion')] = accordion
+        gr.Markdown("💡 **Tip:** You can use [ControlNet-Preprocessors](https://huggingface.co/spaces/RioShiina/ControlNet-Preprocessors) to generate preprocessed images.")
         
         cn_rows, images, series, types, strengths, filepaths = [], [], [], [], [], []
         components.update({
@@ -192,6 +192,7 @@ def create_diffsynth_controlnet_ui(prefix: str, max_units=MAX_CONTROLNETS):
 
     with gr.Accordion("DiffSynth ControlNet Settings", open=False, visible=('controlnet_model_patch' in default_enabled_chains)) as accordion:
         components[key('diffsynth_controlnet_accordion')] = accordion
+        gr.Markdown("💡 **Tip:** You can use [ControlNet-Preprocessors](https://huggingface.co/spaces/RioShiina/ControlNet-Preprocessors) to generate preprocessed images.")
         
         cn_rows, images, series, types, strengths, filepaths = [], [], [], [], [], []
         components.update({
@@ -237,7 +238,7 @@ def create_ipadapter_ui(prefix: str, max_units=MAX_IPADAPTERS):
 
     with gr.Accordion("IPAdapter Settings", open=False, visible=('ipadapter' in default_enabled_chains)) as accordion:
         components[key('ipadapter_accordion')] = accordion
-        gr.Markdown("Powered by [cubiq/ComfyUI_IPAdapter_plus](https://github.com/cubiq/ComfyUI_IPAdapter_plus).")
+        gr.Markdown("💡 **Tip:** Processed using the [cubiq/ComfyUI_IPAdapter_plus](https://github.com/cubiq/ComfyUI_IPAdapter_plus) node.")
         
         with gr.Row():
             components[key('ipadapter_final_preset')] = gr.Dropdown(
@@ -306,6 +307,7 @@ def create_flux1_ipadapter_ui(prefix: str, max_units=MAX_IPADAPTERS):
 
     with gr.Accordion("IPAdapter Settings (FLUX.1)", open=False, visible=('flux1_ipadapter' in default_enabled_chains)) as accordion:
         components[key('flux1_ipadapter_accordion')] = accordion
+        gr.Markdown("💡 **Tip:** Processed using the [Shakker-Labs/ComfyUI-IPAdapter-Flux](https://github.com/Shakker-Labs/ComfyUI-IPAdapter-Flux) node.")
         
         ipa_rows, images, weights, start_percents, end_percents = [], [], [], [], []
         components.update({
@@ -343,6 +345,7 @@ def create_sd3_ipadapter_ui(prefix: str, max_units=MAX_IPADAPTERS):
 
     with gr.Accordion("IPAdapter Settings (SD3)", open=False, visible=('sd3_ipadapter' in default_enabled_chains)) as accordion:
         components[key('sd3_ipadapter_accordion')] = accordion
+        gr.Markdown("💡 **Tip:** Processed using the [Slickytail/ComfyUI-InstantX-IPAdapter-SD3](https://github.com/Slickytail/ComfyUI-InstantX-IPAdapter-SD3) node.")
         
         ipa_rows, images, weights, start_percents, end_percents = [], [], [], [], []
         components.update({

yaml/file_list.yaml

@@ -56,9 +56,6 @@ file:
       source: hf
       repo_id: "Yevrey921/novaAnimeXL_ilV160"
       repository_file_path: "novaAnimeXL_ilV160.safetensors"
-    - filename: "Illustrious-XL-v2.0.safetensors"
-      source: hf
-      repo_id: "OnomaAIResearch/Illustrious-XL-v2.0"
     - filename: "Illustrious-XL-v2.0.safetensors"
       source: hf
       repo_id: "OnomaAIResearch/Illustrious-XL-v2.0"

yaml/model_list.yaml

@@ -6,4 +6,297 @@ Checkpoint:
         components:
           unet: "flux-2-klein-9b-kv-fp8.safetensors"
           clip: "qwen_3_8b_fp8mixed.safetensors"
-          vae: "flux2-vae.safetensors"
+          vae: "flux2-vae.safetensors"
+  FLUX.2:
+    latent_type: flux2_latent
+    models:
+      - display_name: "black-forest-labs/FLUX.2-klein-4B"
+        components:
+          unet: "flux-2-klein-4b-fp8.safetensors"
+          clip: "qwen_3_4b.safetensors"
+          vae: "flux2-vae.safetensors"
+      - display_name: "black-forest-labs/FLUX.2-klein-9B"
+        components:
+          unet: "flux-2-klein-9b-fp8.safetensors"
+          clip: "qwen_3_8b_fp8mixed.safetensors"
+          vae: "flux2-vae.safetensors"
+      - display_name: "black-forest-labs/FLUX.2-klein-base-4B"
+        components:
+          unet: "flux-2-klein-base-4b-fp8.safetensors"
+          clip: "qwen_3_4b.safetensors"
+          vae: "flux2-vae.safetensors"
+      - display_name: "black-forest-labs/FLUX.2-klein-base-9B"
+        components:
+          unet: "flux-2-klein-base-9b-fp8.safetensors"
+          clip: "qwen_3_8b_fp8mixed.safetensors"
+          vae: "flux2-vae.safetensors"
+      - display_name: "black-forest-labs/FLUX.2-dev (Need to set ZeroGPU Duration to 120)"
+        components:
+          unet: "flux2_dev_fp8mixed.safetensors"
+          clip: "mistral_3_small_flux2_fp8.safetensors"
+          vae: "flux2-vae.safetensors"
+  ERNIE-Image:
+    latent_type: flux2_latent
+    models:
+      - display_name: "baidu/ERNIE-Image"
+        components:
+          unet: "ernie-image.safetensors"
+          vae: "flux2-vae.safetensors"
+          clip: "ministral-3-3b.safetensors"
+      - display_name: "baidu/ERNIE-Image-Turbo"
+        components:
+          unet: "ernie-image-turbo.safetensors"
+          vae: "flux2-vae.safetensors"
+          clip: "ministral-3-3b.safetensors"
+  Z-Image:
+    latent_type: sd3_latent
+    models:
+      - display_name: "Tongyi-MAI/Z Image Turbo"
+        components:
+          unet: "z_image_turbo_bf16.safetensors"
+          vae: "ae.safetensors"
+          clip: "qwen_3_4b_fp8_mixed.safetensors"
+      - display_name: "Tongyi-MAI/Z Image"
+        components:
+          unet: "z_image_bf16.safetensors"
+          vae: "ae.safetensors"
+          clip: "qwen_3_4b_fp8_mixed.safetensors"
+  Qwen-Image:
+    latent_type: sd3_latent
+    models:
+      - display_name: "Qwen/Qwen-Image-2512 + Lightning-4steps-V1.0 LoRA"
+        components:
+          unet: "qwen_image_2512_fp8_e4m3fn.safetensors"
+          vae: "qwen_image_vae.safetensors"
+          clip: "qwen_2.5_vl_7b_fp8_scaled.safetensors"
+          lora: "Qwen-Image-2512-Lightning-4steps-V1.0-bf16.safetensors"
+      - display_name: "Qwen/Qwen-Image + Lightning-4steps-V1.0 LoRA"
+        components:
+          unet: "qwen_image_fp8_e4m3fn.safetensors"
+          vae: "qwen_image_vae.safetensors"
+          clip: "qwen_2.5_vl_7b_fp8_scaled.safetensors"
+          lora: "Qwen-Image-fp8-e4m3fn-Lightning-4steps-V1.0-bf16.safetensors"
+  LongCat-Image:
+    latent_type: sd3_latent
+    models:
+      - display_name: "meituan-longcat/LongCat-Image"
+        components:
+          unet: "longcat_image_bf16.safetensors"
+          vae: "ae.safetensors"
+          clip: "qwen_2.5_vl_7b_fp8_scaled.safetensors"
+  Anima:
+    latent_type: latent
+    models:
+      - display_name: "duongve/AnimaYume-v0.4"
+        components:
+          unet: "AnimaYume_tuned_v04.safetensors"
+          vae: "qwen_image_vae.safetensors"
+          clip: "qwen_3_06b_base.safetensors"
+      - display_name: "circlestone-labs/Anima-preview3-base"
+        components:
+          unet: "anima-preview3-base.safetensors"
+          vae: "qwen_image_vae.safetensors"
+          clip: "qwen_3_06b_base.safetensors"
+  NewBie-Image:
+    latent_type: sd3_latent
+    models:
+      - display_name: "NewBie-AI/NewBie-image-Exp0.1"
+        components:
+          unet: "NewBie-Image-Exp0.1-bf16.safetensors"
+          vae: "ae.safetensors"
+          clip1: "gemma_3_4b_it_bf16.safetensors"
+          clip2: "jina_clip_v2_bf16.safetensors"
+  Ovis-Image:
+    latent_type: sd3_latent
+    models:
+      - display_name: "AIDC-AI/Ovis-Image"
+        components:
+          unet: "ovis_image_bf16.safetensors"
+          vae: "ae.safetensors"
+          clip: "ovis_2.5.safetensors"
+  HunyuanImage:
+    latent_type: hunyuan_latent
+    models:
+      - display_name: "HunyuanImage-2.1"
+        components:
+          unet: "hunyuanimage2.1_fp8_e4m3fn.safetensors"
+          vae: "hunyuan_image_2.1_vae_fp16.safetensors"
+          clip1: "qwen_2.5_vl_7b_fp8_scaled.safetensors"
+          clip2: "byt5_small_glyphxl_fp16.safetensors"
+      - display_name: "HunyuanImage-2.1-Distilled"
+        components:
+          unet: "hunyuanimage2.1_distilled_fp8_e4m3fn.safetensors"
+          vae: "hunyuan_image_2.1_vae_fp16.safetensors"
+          clip1: "qwen_2.5_vl_7b_fp8_scaled.safetensors"
+          clip2: "byt5_small_glyphxl_fp16.safetensors"
+  Chroma1-Radiance:
+    latent_type: chroma_radiance_latent
+    models:
+      - display_name: "lodestones/Chroma1-Radiance-x0"
+        components:
+          unet: "Chroma1-Radiance-x0-fp8mixed_fullmm-20260104.safetensors"
+          vae: "pixel_space"
+          clip: "t5xxl_fp8_e4m3fn_scaled.safetensors"
+  Chroma1:
+    latent_type: sd3_latent
+    models:
+      - display_name: "lodestones/Chroma1-HD"
+        components:
+          unet: "Chroma1-HD_float8_e4m3fn_scaled_learned_topk8_svd.safetensors"
+          vae: "ae.safetensors"
+          clip: "t5xxl_fp8_e4m3fn_scaled.safetensors"
+      - display_name: "lodestones/Chroma1-HD-Flash"
+        components:
+          unet: "Chroma1-HD-Flash_float8_e4m3fn_scaled_learned_topk8_svd.safetensors"
+          vae: "ae.safetensors"
+          clip: "t5xxl_fp8_e4m3fn_scaled.safetensors"
+  OmniGen2:
+    latent_type: sd3_latent
+    models:
+      - display_name: "OmniGen2"
+        components:
+          unet: "omnigen2_fp16.safetensors"
+          vae: "ae.safetensors"
+          clip: "qwen_2.5_vl_fp16.safetensors"
+  Lumina:
+    latent_type: sd3_latent
+    models:
+      - display_name: "Alpha-VLLM/Lumina-Image-2.0"
+        path: "lumina_2.safetensors"
+  HiDream:
+    latent_type: sd3_latent
+    models:
+      - display_name: "HiDream_i1_Dev"
+        components:
+          unet: "hidream_i1_dev_fp8.safetensors"
+          vae: "ae.safetensors"
+          clip1: "clip_l_hidream.safetensors"
+          clip2: "clip_g_hidream.safetensors"
+          clip3: "t5xxl_fp8_e4m3fn_scaled.safetensors"
+          clip4: "llama_3.1_8b_instruct_fp8_scaled.safetensors"
+      - display_name: "HiDream_i1_Full"
+        components:
+          unet: "hidream_i1_full_fp8.safetensors"
+          vae: "ae.safetensors"
+          clip1: "clip_l_hidream.safetensors"
+          clip2: "clip_g_hidream.safetensors"
+          clip3: "t5xxl_fp8_e4m3fn_scaled.safetensors"
+          clip4: "llama_3.1_8b_instruct_fp8_scaled.safetensors"
+      - display_name: "HiDream_i1_Fast"
+        components:
+          unet: "hidream_i1_fast_fp8.safetensors"
+          vae: "ae.safetensors"
+          clip1: "clip_l_hidream.safetensors"
+          clip2: "clip_g_hidream.safetensors"
+          clip3: "t5xxl_fp8_e4m3fn_scaled.safetensors"
+          clip4: "llama_3.1_8b_instruct_fp8_scaled.safetensors"
+  FLUX.1:
+    latent_type: sd3_latent
+    models:
+      - display_name: "flux1-dev"
+        components:
+          unet: "flux1-dev-fp8-e4m3fn.safetensors"
+          vae: "ae.safetensors"
+          clip1: "clip_l.safetensors"
+          clip2: "t5xxl_fp8_e4m3fn_scaled.safetensors"
+      - display_name: "flux1-schnell"
+        components:
+          unet: "flux1-schnell-fp8-e4m3fn.safetensors"
+          vae: "ae.safetensors"
+          clip1: "clip_l.safetensors"
+          clip2: "t5xxl_fp8_e4m3fn_scaled.safetensors"
+      - display_name: "flux1-krea-dev"
+        components:
+          unet: "flux1-krea-dev_fp8_scaled.safetensors"
+          vae: "ae.safetensors"
+          clip1: "clip_l.safetensors"
+          clip2: "t5xxl_fp8_e4m3fn_scaled.safetensors"
+  SD3.5:
+    latent_type: sd3_latent
+    models:
+      - display_name: "StabilityAI/SD3.5-Large"
+        path: "sd3.5_large_fp8_scaled.safetensors"
+      - display_name: "StabilityAI/SD3.5-Medium"
+        path: "sd3.5_medium_incl_clips_t5xxlfp8scaled.safetensors"
+  SDXL:
+    latent_type: latent
+    models:
+      # NoobAI
+      - display_name: "Laxhar/noobai-XL-Vpred-1.0"
+        path: "NoobAI-XL-Vpred-v1.0.safetensors"
+        category: "NoobAI"
+      - display_name: "Laxhar/noobai-XL-1.1"
+        path: "NoobAI-XL-v1.1.safetensors"
+        category: "NoobAI"
+      - display_name: "bluepen5805/noob_v_pencil-XL-3.0.0"
+        path: "noob_v_pencil-XL-v3.0.0.safetensors"
+        category: "NoobAI"
+      - display_name: "RedRayz/hikari_noob_v-pred_1.2.4"
+        path: "Hikari_Noob_v-pred_1.2.4.safetensors"
+        category: "NoobAI"
+      - display_name: "ChenkinNoob/ChenkinNoob-XL-V0.5"
+        path: "ChenkinNoob-XL-V0.5.safetensors"
+        category: "NoobAI"
+      # Illustrious
+      - display_name: "WAI0731/wai-illustrious-sdxl-v170"
+        path: "waiIllustriousSDXL_v170.safetensors"
+        category: "Illustrious"
+      - display_name: "blue_pen5805/mellow_pencil-XL-v1.0.0"
+        path: "mellow_pencil-XL-v1.0.0.safetensors"
+        category: "Illustrious"
+      - display_name: "blue_pen5805/illustrious_pencil-XL-v5.0.0"
+        path: "illustrious_pencil-XL-v5.0.0.safetensors"
+        category: "Illustrious"
+      - display_name: "Ikena/hassaku-xl-illustrious-v3.4"
+        path: "hassakuXLIllustrious_v34.safetensors"
+        category: "Illustrious"
+      - display_name: "Crody/Nova-Anime-XL-v16.0"
+        path: "novaAnimeXL_ilV160.safetensors"
+        category: "Illustrious"
+      - display_name: "OnomaAIResearch/Illustrious-XL-v2.0"
+        path: "Illustrious-XL-v2.0.safetensors"
+        category: "Illustrious"
+      - display_name: "OnomaAIResearch/Illustrious-XL-v1.1"
+        path: "Illustrious-XL-v1.1.safetensors"
+        category: "Illustrious"
+      - display_name: "OnomaAIResearch/Illustrious-XL-v1.0"
+        path: "Illustrious-XL-v1.0.safetensors"
+        category: "Illustrious"
+      - display_name: "aria1th261/Illustrious-XL-v0.1"
+        path: "illustriousXL_v01.safetensors"
+        category: "Illustrious"
+    # SDXL-Animate
+      - display_name: "CagliostroLab/Animagine XL 4.0"
+        path: "animagine-xl-4.0.safetensors"
+        category: "Animate"
+      - display_name: "CagliostroLab/Animagine XL 3.1"
+        path: "animagine-xl-3.1.safetensors"
+        category: "Animate"
+      - display_name: "blue_pen5805/4nima_pencil-XL v1.0.1"
+        path: "4nima_pencil-XL-v1.0.1.safetensors"
+        category: "Animate"
+      - display_name: "blue_pen5805/anima_pencil-XL v5.0.0"
+        path: "anima_pencil-XL-v5.0.0.safetensors"
+        category: "Animate"
+      - display_name: "blue_pen5805/blue_pencil-XL v7.0.0"
+        path: "blue_pencil-XL-v7.0.0.safetensors"
+        category: "Animate"
+    # SDXL-Pony
+      - display_name: "PurpleSmartAI/Pony Diffusion V6 XL"
+        path: "ponyDiffusionV6XL_v6StartWithThisOne.safetensors"
+        category: "Pony"
+      - display_name: "blue_pen5805/pony_pencil-XL v2.0.0"
+        path: "pony_pencil-XL-v2.0.0.safetensors"
+        category: "Pony"
+      - display_name: "Cyberdelia/CyberRealistic Pony v14.0"
+        path: "CyberRealisticPony_V14.0.safetensors"
+        category: "Pony"
+    # SDXL-Base
+      - display_name: "stabilityai/stable-diffusion-xl-base-1.0"
+        path: "sd_xl_base_1.0.safetensors"
+        category: "Base"
+  SD1.5:
+    latent_type: latent
+    models:
+      - display_name: "stable-diffusion-v1-5/stable-diffusion-v1-5"
+        path: "v1-5-pruned-emaonly.safetensors"