Upload folder using huggingface_hub

README.md

@@ -6,6 +6,7 @@ colorTo: red
 sdk: gradio
 sdk_version: "5.50.0"
 app_file: app.py
+python_version: 3.12
 short_description: Multi-task image generator with dynamic, chainable workflows
 pinned: true
 models:
@@ -18,6 +19,7 @@ models:
   - black-forest-labs/FLUX.2-klein-base-4b-fp8
   - black-forest-labs/FLUX.2-klein-base-9b-fp8
   - bluepen5805/4nima_pencil-XL
+  - bluepen5805/anima-models
   - bluepen5805/anima_pencil-XL
   - bluepen5805/blue_pencil-XL
   - bluepen5805/illustrious_pencil-XL

chain_injectors/hidream_o1_reference_injector.py

@@ -0,0 +1,54 @@
+def inject(assembler, chain_definition, chain_items):
+    if not chain_items:
+        return
+
+    ksampler_name = chain_definition.get('ksampler_node', 'ksampler')
+
+    if ksampler_name not in assembler.node_map:
+        print(f"Warning: KSampler node '{ksampler_name}' not found for HiDream-O1 Reference chain. Skipping.")
+        return
+
+    ksampler_id = assembler.node_map[ksampler_name]
+
+    if 'positive' not in assembler.workflow[ksampler_id]['inputs'] or 'negative' not in assembler.workflow[ksampler_id]['inputs']:
+        print(f"Warning: KSampler node '{ksampler_name}' missing positive/negative inputs. Skipping.")
+        return
+
+    current_pos_conditioning = assembler.workflow[ksampler_id]['inputs']['positive']
+    current_neg_conditioning = assembler.workflow[ksampler_id]['inputs']['negative']
+
+    ref_images_id = assembler._get_unique_id()
+    ref_images_node = assembler._get_node_template("HiDreamO1ReferenceImages")
+    
+    if 'images' in ref_images_node['inputs']:
+        del ref_images_node['inputs']['images']
+        
+    ref_images_node['inputs']['positive'] = current_pos_conditioning
+    ref_images_node['inputs']['negative'] = current_neg_conditioning
+
+    for i, img_filename in enumerate(chain_items):
+        if i >= 10:
+            break
+            
+        load_id = assembler._get_unique_id()
+        load_node = assembler._get_node_template("LoadImage")
+        load_node['inputs']['image'] = img_filename
+        load_node['_meta']['title'] = f"Load Reference Image {i+1}"
+        assembler.workflow[load_id] = load_node
+        
+        scale_id = assembler._get_unique_id()
+        scale_node = assembler._get_node_template("ImageScaleToTotalPixels")
+        scale_node['inputs']['megapixels'] = 1.0
+        scale_node['inputs']['upscale_method'] = "lanczos"
+        scale_node['inputs']['image'] = [load_id, 0]
+        scale_node['_meta']['title'] = f"Scale Reference {i+1}"
+        assembler.workflow[scale_id] = scale_node
+        
+        ref_images_node['inputs'][f'images.image_{i+1}'] = [scale_id, 0]
+
+    assembler.workflow[ref_images_id] = ref_images_node
+
+    assembler.workflow[ksampler_id]['inputs']['positive'] = [ref_images_id, 0]
+    assembler.workflow[ksampler_id]['inputs']['negative'] = [ref_images_id, 1]
+    
+    print(f"HiDream-O1 Reference injector applied. Re-routed inputs through {min(len(chain_items), 10)} reference images.")

chain_injectors/{hidream_smoothing_injector.py → hidream_o1_smoothing_injector.py}

@@ -6,7 +6,7 @@ def inject(assembler, chain_definition, chain_items):
 
     target_node_name = chain_definition.get('target_node')
     if not target_node_name or target_node_name not in assembler.node_map:
-        print(f"Warning: Target node '{target_node_name}' not found for HiDream Smoothing. Skipping.")
+        print(f"Warning: Target node '{target_node_name}' not found for HiDream-O1 Smoothing. Skipping.")
         return
 
     target_node_id = assembler.node_map[target_node_name]

core/pipelines/sd_image_pipeline.py

@@ -96,9 +96,17 @@ class SdImagePipeline(BasePipeline):
                             raise RuntimeError(f"Workflow integrity error: Output of node {source_node_id} needed for {node_id} but not yet computed.")
                         
                         source_output_tuple = computed_outputs[source_node_id]
-                        kwargs[param_name] = get_value_at_index(source_output_tuple, output_index)
+                        actual_value = get_value_at_index(source_output_tuple, output_index)
                     else:
-                        kwargs[param_name] = param_value
+                        actual_value = param_value
+
+                    if '.' in param_name:
+                        parent_key, child_key = param_name.split('.', 1)
+                        if parent_key not in kwargs or not isinstance(kwargs[parent_key], dict):
+                            kwargs[parent_key] = {}
+                        kwargs[parent_key][child_key] = actual_value
+                    else:
+                        kwargs[param_name] = actual_value
 
                 function_name = getattr(node_class, 'FUNCTION')
                 execution_method = getattr(node_instance, function_name)
@@ -436,6 +444,17 @@ class SdImagePipeline(BasePipeline):
                     img.save(temp_path, "PNG")
                     temp_files_to_clean.append(temp_path)
                     active_reference_latents.append(os.path.basename(temp_path))
+                    
+        hidream_o1_reference_data = ui_inputs.get('hidream_o1_reference_data', [])
+        active_hidream_o1_reference = []
+        if hidream_o1_reference_data:
+            for img in hidream_o1_reference_data:
+                if img:
+                    if not os.path.exists(INPUT_DIR): os.makedirs(INPUT_DIR)
+                    temp_path = os.path.join(INPUT_DIR, f"temp_ho1_ref_{random.randint(1000, 9999)}.png")
+                    img.save(temp_path, "PNG")
+                    temp_files_to_clean.append(temp_path)
+                    active_hidream_o1_reference.append(os.path.basename(temp_path))
         
         from utils.app_utils import get_vae_path
         vae_source = ui_inputs.get('vae_source')
@@ -491,9 +510,9 @@ class SdImagePipeline(BasePipeline):
         recipe_path = os.path.join(os.path.dirname(__file__), "workflow_recipes", "sd_unified_recipe.yaml")
         assembler = WorkflowAssembler(recipe_path, dynamic_values=dynamic_values)
         
-        hidream_smoothing_data = []
+        hidream_o1_smoothing_data = []
         if workflow_model_type == 'hidream-o1' and model_display_name == "HiDream-O1-Image":
-            hidream_smoothing_data.append({})
+            hidream_o1_smoothing_data.append({})
 
         workflow_inputs = {
             **ui_inputs,
@@ -515,8 +534,9 @@ class SdImagePipeline(BasePipeline):
             "style_chain": active_styles,
             "conditioning_chain": active_conditioning,
             "reference_latent_chain": active_reference_latents,
+            "hidream_o1_reference_chain": active_hidream_o1_reference,
             "vae_chain": [ui_inputs.get('vae_name')] if ui_inputs.get('vae_name') else [],
-            "hidream_smoothing_chain": hidream_smoothing_data,
+            "hidream_o1_smoothing_chain": hidream_o1_smoothing_data,
         }
 
         if isinstance(path_or_components, dict):

core/pipelines/workflow_recipes/_partials/conditioning/hidream-o1.yaml

@@ -51,8 +51,8 @@ dynamic_lora_chains:
       "model": ["model_noise_scale:model"]
       "clip": ["pos_prompt:clip", "neg_prompt:clip"]
 
-dynamic_hidream_smoothing_chains:
-  hidream_smoothing_chain:
+dynamic_hidream_o1_smoothing_chains:
+  hidream_o1_smoothing_chain:
     target_node: "ksampler"
 
 dynamic_conditioning_chains:
@@ -60,5 +60,9 @@ dynamic_conditioning_chains:
     ksampler_node: "ksampler"
     clip_source: "ckpt_loader:1"
 
+dynamic_hidream_o1_reference_chains:
+  hidream_o1_reference_chain:
+    ksampler_node: "ksampler"
+
 ui_map:
   model_name: "ckpt_loader:ckpt_name"

requirements.txt

@@ -1,5 +1,5 @@
-comfyui-frontend-package==1.43.18
-comfyui-workflow-templates==0.9.77
+comfyui-frontend-package==1.44.19
+comfyui-workflow-templates==0.9.82
 comfyui-embedded-docs==0.5.0
 torch
 torchsde
@@ -23,7 +23,7 @@ SQLAlchemy>=2.0.0
 filelock
 av>=14.2.0
 comfy-kitchen>=0.2.8
-comfy-aimdo==0.3.0
+comfy-aimdo==0.4.5
 requests
 simpleeval>=1.0.0
 blake3
@@ -43,4 +43,4 @@ huggingface-hub
 imageio
 
 spaces
-sageattention @ https://huggingface.co/RioShiina/Sage-Attention-ZeroGPU-Space-Build/resolve/main/sageattention-2.2.0-cp310-cp310-linux_x86_64.whl
+sageattention @ https://huggingface.co/RioShiina/Sage-Attention-ZeroGPU-Space-Build/resolve/main/sageattention-2.2.0-cp312-cp312-linux_x86_64.whl

ui/events.py

@@ -605,6 +605,44 @@ def attach_event_handlers(ui_components, demo):
         add_button.click(fn=add_fipa_row, inputs=[count_state], outputs=add_outputs, show_progress=False)
         del_button.click(fn=del_fipa_row, inputs=[count_state], outputs=del_outputs, show_progress=False)
 
+    def create_sd3_ipadapter_event_handlers(prefix):
+        ipa_rows = ui_components.get(f'sd3_ipadapter_rows_{prefix}')
+        if not ipa_rows: return
+        count_state = ui_components[f'sd3_ipadapter_count_state_{prefix}']
+        add_button = ui_components[f'add_sd3_ipadapter_button_{prefix}']
+        del_button = ui_components[f'delete_sd3_ipadapter_button_{prefix}']
+        images = ui_components[f'sd3_ipadapter_images_{prefix}']
+        weights = ui_components[f'sd3_ipadapter_weights_{prefix}']
+        start_percents = ui_components[f'sd3_ipadapter_start_percents_{prefix}']
+        end_percents = ui_components[f'sd3_ipadapter_end_percents_{prefix}']
+
+        def add_ipa_row(c):
+            c += 1
+            return {
+                count_state: c,
+                ipa_rows[c - 1]: gr.update(visible=True),
+                add_button: gr.update(visible=c < MAX_IPADAPTERS),
+                del_button: gr.update(visible=True),
+            }
+
+        def del_ipa_row(c):
+            c -= 1
+            return {
+                count_state: c,
+                ipa_rows[c]: gr.update(visible=False),
+                images[c]: None,
+                weights[c]: 0.5,
+                start_percents[c]: 0.0,
+                end_percents[c]: 1.0,
+                add_button: gr.update(visible=True),
+                del_button: gr.update(visible=c > 0),
+            }
+
+        add_outputs = [count_state, add_button, del_button] + ipa_rows
+        del_outputs = [count_state, add_button, del_button] + ipa_rows + images + weights + start_percents + end_percents
+        add_button.click(fn=add_ipa_row, inputs=[count_state], outputs=add_outputs, show_progress=False)
+        del_button.click(fn=del_ipa_row, inputs=[count_state], outputs=del_outputs, show_progress=False)
+
     def create_style_event_handlers(prefix):
         style_rows = ui_components.get(f'style_rows_{prefix}')
         if not style_rows: return
@@ -717,6 +755,37 @@ def attach_event_handlers(ui_components, demo):
         add_button.click(fn=add_ref_row, inputs=[count_state], outputs=add_outputs, show_progress=False)
         del_button.click(fn=del_ref_row, inputs=[count_state], outputs=del_outputs, show_progress=False)
 
+    def create_hidream_o1_reference_event_handlers(prefix):
+        ref_rows = ui_components.get(f'hidream_o1_reference_rows_{prefix}')
+        if not ref_rows: return
+        count_state = ui_components[f'hidream_o1_reference_count_state_{prefix}']
+        add_button = ui_components[f'add_hidream_o1_reference_button_{prefix}']
+        del_button = ui_components[f'delete_hidream_o1_reference_button_{prefix}']
+        images = ui_components[f'hidream_o1_reference_images_{prefix}']
+
+        def add_ref_row(c):
+            c += 1
+            return {
+                count_state: c,
+                ref_rows[c - 1]: gr.update(visible=True),
+                add_button: gr.update(visible=c < 10),
+                del_button: gr.update(visible=True),
+            }
+
+        def del_ref_row(c):
+            c -= 1
+            return {
+                count_state: c,
+                ref_rows[c]: gr.update(visible=False),
+                images[c]: None,
+                add_button: gr.update(visible=True),
+                del_button: gr.update(visible=c > 0),
+            }
+
+        add_outputs = [count_state, add_button, del_button] + ref_rows
+        del_outputs = [count_state, add_button, del_button] + ref_rows + images
+        add_button.click(fn=add_ref_row, inputs=[count_state], outputs=add_outputs, show_progress=False)
+        del_button.click(fn=del_ref_row, inputs=[count_state], outputs=del_outputs, show_progress=False)
 
     def create_embedding_event_handlers(prefix):
         rows = ui_components.get(f'embedding_rows_{prefix}')
@@ -852,6 +921,7 @@ def attach_event_handlers(ui_components, demo):
         embedding_data_components = ui_components.get(f'all_embedding_components_flat_{prefix}', [])
         conditioning_data_components = ui_components.get(f'all_conditioning_components_flat_{prefix}', [])
         reference_latent_data_components = ui_components.get(f'all_reference_latent_components_flat_{prefix}', [])
+        hidream_o1_reference_data_components = ui_components.get(f'all_hidream_o1_reference_components_flat_{prefix}', [])
         
         run_inputs_map['vae_source'] = ui_components.get(f'vae_source_{prefix}')
         run_inputs_map['vae_id'] = ui_components.get(f'vae_id_{prefix}')
@@ -862,7 +932,7 @@ def attach_event_handlers(ui_components, demo):
         all_chains = [
             lora_data_components, controlnet_data_components, anima_controlnet_lllite_data_components, diffsynth_controlnet_data_components, ipadapter_data_components,
             sd3_ipadapter_data_components, flux1_ipadapter_data_components, style_data_components,
-            embedding_data_components, conditioning_data_components, reference_latent_data_components
+            embedding_data_components, conditioning_data_components, reference_latent_data_components, hidream_o1_reference_data_components
         ]
         for chain in all_chains:
             if chain:
@@ -890,6 +960,7 @@ def attach_event_handlers(ui_components, demo):
             assign_chain_data('embedding_data', embedding_data_components)
             assign_chain_data('conditioning_data', conditioning_data_components)
             assign_chain_data('reference_latent_data', reference_latent_data_components)
+            assign_chain_data('hidream_o1_reference_data', hidream_o1_reference_data_components)
 
             return ui_dict
 
@@ -902,7 +973,7 @@ def attach_event_handlers(ui_components, demo):
                 outputs=[res_gal]
             )
 
-    def make_update_fn(m_comp, cat_comp, cs_comp, ar_comp, width_comp, height_comp, cn_types, cn_series, cn_filepaths, anima_cn_types, anima_cn_series, anima_cn_filepaths, diffsynth_cn_types, diffsynth_cn_series, diffsynth_cn_filepaths, ipa_preset, lora_acc, cn_acc, anima_cn_acc, diffsynth_cn_acc, ipa_acc, sd3_ipa_acc, flux1_ipa_acc, style_acc, embed_acc, cond_acc, ref_latent_acc, guidance_comp, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp):
+    def make_update_fn(m_comp, cat_comp, cs_comp, ar_comp, width_comp, height_comp, cn_types, cn_series, cn_filepaths, anima_cn_types, anima_cn_series, anima_cn_filepaths, diffsynth_cn_types, diffsynth_cn_series, diffsynth_cn_filepaths, ipa_preset, lora_acc, cn_acc, anima_cn_acc, diffsynth_cn_acc, ipa_acc, sd3_ipa_acc, flux1_ipa_acc, style_acc, embed_acc, cond_acc, ref_latent_acc, hidream_o1_ref_acc, guidance_comp, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp):
         def update_fn(*args):
             arch = args[0]
             category = args[1]
@@ -953,6 +1024,7 @@ def attach_event_handlers(ui_components, demo):
             if embed_acc: updates[embed_acc] = gr.update(visible=('embedding' in enabled_chains))
             if cond_acc: updates[cond_acc] = gr.update(visible=('conditioning' in enabled_chains))
             if ref_latent_acc: updates[ref_latent_acc] = gr.update(visible=('reference_latent' in enabled_chains))
+            if hidream_o1_ref_acc: updates[hidream_o1_ref_acc] = gr.update(visible=('hidream_o1_reference' in enabled_chains))
 
             if cs_comp:
                 updates[cs_comp] = gr.update(visible=(arch_model_type == "sd15"))
@@ -1016,7 +1088,7 @@ def attach_event_handlers(ui_components, demo):
             return updates
         return update_fn
 
-    def make_model_change_fn(cat_comp_ref, cs_comp, ar_comp, width_comp, height_comp, cn_types, cn_series, cn_filepaths, anima_cn_types, anima_cn_series, anima_cn_filepaths, diffsynth_cn_types, diffsynth_cn_series, diffsynth_cn_filepaths, arch_comp_ref, ipa_preset, lora_acc, cn_acc, anima_cn_acc, diffsynth_cn_acc, ipa_acc, sd3_ipa_acc, flux1_ipa_acc, style_acc, embed_acc, cond_acc, ref_latent_acc, guidance_comp, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp):
+    def make_model_change_fn(cat_comp_ref, cs_comp, ar_comp, width_comp, height_comp, cn_types, cn_series, cn_filepaths, anima_cn_types, anima_cn_series, anima_cn_filepaths, diffsynth_cn_types, diffsynth_cn_series, diffsynth_cn_filepaths, arch_comp_ref, ipa_preset, lora_acc, cn_acc, anima_cn_acc, diffsynth_cn_acc, ipa_acc, sd3_ipa_acc, flux1_ipa_acc, style_acc, embed_acc, cond_acc, ref_latent_acc, hidream_o1_ref_acc, guidance_comp, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp):
         def change_fn(*args):
             model_name = args[0]
             idx = 1
@@ -1071,6 +1143,7 @@ def attach_event_handlers(ui_components, demo):
             if embed_acc: updates[embed_acc] = gr.update(visible=('embedding' in enabled_chains))
             if cond_acc: updates[cond_acc] = gr.update(visible=('conditioning' in enabled_chains))
             if ref_latent_acc: updates[ref_latent_acc] = gr.update(visible=('reference_latent' in enabled_chains))
+            if hidream_o1_ref_acc: updates[hidream_o1_ref_acc] = gr.update(visible=('hidream_o1_reference' in enabled_chains))
 
             if cs_comp:
                 updates[cs_comp] = gr.update(visible=(arch_model_type == "sd15"))
@@ -1172,6 +1245,7 @@ def attach_event_handlers(ui_components, demo):
         embedding_accordion = ui_components.get(f'embedding_accordion_{prefix}')
         conditioning_accordion = ui_components.get(f'conditioning_accordion_{prefix}')
         ref_latent_accordion = ui_components.get(f'reference_latent_accordion_{prefix}')
+        hidream_o1_ref_accordion = ui_components.get(f'hidream_o1_reference_accordion_{prefix}')
         
         ipa_preset_list = ui_components.get(f'ipadapter_final_preset_{prefix}')
         
@@ -1204,6 +1278,7 @@ def attach_event_handlers(ui_components, demo):
             if embedding_accordion: outputs.append(embedding_accordion)
             if conditioning_accordion: outputs.append(conditioning_accordion)
             if ref_latent_accordion: outputs.append(ref_latent_accordion)
+            if hidream_o1_ref_accordion: outputs.append(hidream_o1_ref_accordion)
             if ipa_preset_list: outputs.append(ipa_preset_list)
             
             outputs.extend(valid_extra_comps)
@@ -1214,7 +1289,7 @@ def attach_event_handlers(ui_components, demo):
                 anima_cn_types_list, anima_cn_series_list, anima_cn_filepaths_list,
                 diffsynth_cn_types_list, diffsynth_cn_series_list, diffsynth_cn_filepaths_list, 
                 ipa_preset_list, lora_accordion, cn_accordion, anima_cn_accordion, diffsynth_cn_accordion, ipa_accordion, sd3_ipa_accordion, flux1_ipa_accordion, style_accordion, embedding_accordion, conditioning_accordion,
-                ref_latent_accordion, guidance_comp, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp
+                ref_latent_accordion, hidream_o1_ref_accordion, guidance_comp, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp
             )
             inputs = [arch_comp, cat_comp]
             if aspect_ratio_comp:
@@ -1243,6 +1318,7 @@ def attach_event_handlers(ui_components, demo):
             if embedding_accordion: outputs2.append(embedding_accordion)
             if conditioning_accordion: outputs2.append(conditioning_accordion)
             if ref_latent_accordion: outputs2.append(ref_latent_accordion)
+            if hidream_o1_ref_accordion: outputs2.append(hidream_o1_ref_accordion)
             if ipa_preset_list: outputs2.append(ipa_preset_list)
             
             outputs2.extend(valid_extra_comps)
@@ -1258,7 +1334,7 @@ def attach_event_handlers(ui_components, demo):
                     anima_cn_types_list, anima_cn_series_list, anima_cn_filepaths_list,
                     diffsynth_cn_types_list, diffsynth_cn_series_list, diffsynth_cn_filepaths_list, 
                     arch_comp, ipa_preset_list, lora_accordion, cn_accordion, anima_cn_accordion, diffsynth_cn_accordion, ipa_accordion, sd3_ipa_accordion, flux1_ipa_accordion, style_accordion, embedding_accordion, conditioning_accordion,
-                    ref_latent_accordion, guidance_comp, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp
+                    ref_latent_accordion, hidream_o1_ref_accordion, guidance_comp, prompt_comp, neg_prompt_comp, steps_comp, cfg_comp, sampler_comp, scheduler_comp
                 )
                 model_comp.change(fn=change_fn, inputs=inputs2, outputs=outputs2)
 
@@ -1270,8 +1346,10 @@ def attach_event_handlers(ui_components, demo):
         create_embedding_event_handlers(prefix)
         create_conditioning_event_handlers(prefix)
         create_flux1_ipadapter_event_handlers(prefix)
+        create_sd3_ipadapter_event_handlers(prefix)
         create_style_event_handlers(prefix)
         create_reference_latent_event_handlers(prefix)
+        create_hidream_o1_reference_event_handlers(prefix)
         create_run_event(prefix, task_type)
 
 

ui/shared/hires_fix_ui.py

@@ -7,7 +7,7 @@ from .ui_components import (
     create_conditioning_ui, create_vae_override_ui,
     create_model_architecture_filter_ui, create_category_filter_ui,
     create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
-    create_reference_latent_ui
+    create_reference_latent_ui, create_hidream_o1_reference_ui
 )
 
 def create_ui():
@@ -82,6 +82,7 @@ def create_ui():
         components.update(create_embedding_ui(prefix))
         components.update(create_conditioning_ui(prefix))
         components.update(create_reference_latent_ui(prefix))
+        components.update(create_hidream_o1_reference_ui(prefix))
         components.update(create_vae_override_ui(prefix))
             
     return components

ui/shared/img2img_ui.py

@@ -7,7 +7,7 @@ from .ui_components import (
     create_conditioning_ui, create_vae_override_ui,
     create_model_architecture_filter_ui, create_category_filter_ui,
     create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
-    create_reference_latent_ui
+    create_reference_latent_ui, create_hidream_o1_reference_ui
 )
 
 def create_ui():
@@ -64,6 +64,7 @@ def create_ui():
         components.update(create_style_ui(prefix))
         components.update(create_conditioning_ui(prefix))
         components.update(create_reference_latent_ui(prefix))
+        components.update(create_hidream_o1_reference_ui(prefix))
         components.update(create_vae_override_ui(prefix))
             
     return components

ui/shared/inpaint_ui.py

@@ -6,7 +6,7 @@ from .ui_components import (
     create_conditioning_ui, create_vae_override_ui,
     create_model_architecture_filter_ui, create_category_filter_ui,
     create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
-    create_reference_latent_ui
+    create_reference_latent_ui, create_hidream_o1_reference_ui
 )
 
 def create_ui():
@@ -96,6 +96,7 @@ def create_ui():
             components.update(create_embedding_ui(prefix))
             components.update(create_conditioning_ui(prefix))
             components.update(create_reference_latent_ui(prefix))
+            components.update(create_hidream_o1_reference_ui(prefix))
             components.update(create_vae_override_ui(prefix))
         components[f'accordion_wrapper_{prefix}'] = accordion_wrapper
             

ui/shared/outpaint_ui.py

@@ -7,7 +7,7 @@ from .ui_components import (
     create_conditioning_ui, create_vae_override_ui,
     create_model_architecture_filter_ui, create_category_filter_ui,
     create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
-    create_reference_latent_ui
+    create_reference_latent_ui, create_hidream_o1_reference_ui
 )
 
 def create_ui():
@@ -79,6 +79,7 @@ def create_ui():
         components.update(create_embedding_ui(prefix))
         components.update(create_conditioning_ui(prefix))
         components.update(create_reference_latent_ui(prefix))
+        components.update(create_hidream_o1_reference_ui(prefix))
         components.update(create_vae_override_ui(prefix))
             
     return components

ui/shared/txt2img_ui.py

@@ -6,7 +6,7 @@ from .ui_components import (
     create_conditioning_ui, create_vae_override_ui,
     create_model_architecture_filter_ui, create_category_filter_ui,
     create_sd3_ipadapter_ui, create_flux1_ipadapter_ui, create_style_ui,
-    create_reference_latent_ui
+    create_reference_latent_ui, create_hidream_o1_reference_ui
 )
 
 def create_ui():
@@ -51,6 +51,7 @@ def create_ui():
         components.update(create_style_ui(prefix))
         components.update(create_conditioning_ui(prefix))
         components.update(create_reference_latent_ui(prefix))
+        components.update(create_hidream_o1_reference_ui(prefix))
         components.update(create_vae_override_ui(prefix))
         
     return components

ui/shared/ui_components.py

@@ -535,7 +535,7 @@ def create_conditioning_ui(prefix: str):
         components[key('all_conditioning_components_flat')] = all_cond_components_flat
     
     return components
-    
+
 def on_vae_upload(file_obj):
     if not file_obj:
         return gr.update(), gr.update(), None
@@ -588,7 +588,7 @@ def create_reference_latent_ui(prefix: str, max_units=10):
 
     with gr.Accordion("Reference Edit Settings", open=False, visible=('reference_latent' in default_enabled_chains)) as ref_accordion:
         components[key('reference_latent_accordion')] = ref_accordion
-        gr.Markdown("💡 **Tip:** For multimodal models (like FLUX.2 or OmniGen), this feature enables powerful editing and combining capabilities. In txt2img mode, adding a single reference image performs an **Image Edit**, while adding multiple images performs an **Image Combine**.")
+        gr.Markdown("💡 **Tip:** For multimodal models, this feature enables powerful editing and combining capabilities. In txt2img mode, adding a single reference image performs an **Image Edit**, while adding multiple images performs an **Image Combine**.")
         
         ref_image_groups = []
         ref_image_inputs = []
@@ -609,4 +609,33 @@ def create_reference_latent_ui(prefix: str, max_units=10):
         
         components[key('all_reference_latent_components_flat')] = ref_image_inputs
 
+    return components
+
+def create_hidream_o1_reference_ui(prefix: str, max_units=10):
+    components = {}
+    key = lambda name: f"{name}_{prefix}"
+
+    with gr.Accordion("HiDream-O1 Reference Edit Settings", open=False, visible=('hidream_o1_reference' in default_enabled_chains)) as ref_accordion:
+        components[key('hidream_o1_reference_accordion')] = ref_accordion
+        gr.Markdown("💡 **Tip:** Please use **HiDream-O1-Image-Dev** (HiDream-O1-Image will time out), and set the resolution to **2.0MP** (e.g., 2048x2048). In txt2img mode, adding a single reference image performs an **Image Edit**, while adding multiple images performs an **Image Combine**.")
+        
+        ref_image_groups = []
+        ref_image_inputs = []
+        with gr.Row():
+            for i in range(max_units):
+                with gr.Column(visible=(i < 1), min_width=160) as img_col:
+                    img_comp = gr.Image(type="pil", label=f"Ref. {i+1}", sources=["upload"], height=150)
+                    ref_image_groups.append(img_col)
+                    ref_image_inputs.append(img_comp)
+        
+        components[key('hidream_o1_reference_rows')] = ref_image_groups
+        components[key('hidream_o1_reference_images')] = ref_image_inputs
+
+        with gr.Row():
+            components[key('add_hidream_o1_reference_button')] = gr.Button("✚ Add Reference Image")
+            components[key('delete_hidream_o1_reference_button')] = gr.Button("➖ Delete Reference Image", visible=False)
+        components[key('hidream_o1_reference_count_state')] = gr.State(1)
+        
+        components[key('all_hidream_o1_reference_components_flat')] = ref_image_inputs
+
     return components

yaml/file_list.yaml

@@ -412,10 +412,14 @@ file:
       source: "hf"
       repo_id: "circlestone-labs/Anima"
       repository_file_path: "split_files/diffusion_models/anima-base-v1.0.safetensors"
-    - filename: "AnimaYume_tuned_v04.safetensors"
+    - filename: "AnimaYume_tuned_v05.safetensors"
       source: "hf"
       repo_id: "duongve/AnimaYume"
-      repository_file_path: "split_files/diffusion_models/AnimaYume_tuned_v04.safetensors"
+      repository_file_path: "split_files/diffusion_models/AnimaYume_tuned_v05.safetensors"
+    - filename: "anima_pencil-v1.0.0.safetensors"
+      source: "hf"
+      repo_id: "bluepen5805/anima-models"
+      repository_file_path: "anima_pencil-v1.0.0.safetensors"
     # NewBie-Image
     - filename: "NewBie-Image-Exp0.1-bf16.safetensors"
       source: "hf"

yaml/image_gen_features.yaml

@@ -104,6 +104,7 @@ hidream-o1:
   enabled_chains:
   - lora
   - conditioning
+  - hidream_o1_reference
   
 hidream-i1:
   enabled_chains:

yaml/injectors.yaml

@@ -23,8 +23,10 @@ injector_definitions:
     module: "chain_injectors.conditioning_injector"
   dynamic_style_chains:
     module: "chain_injectors.style_injector"
-  dynamic_hidream_smoothing_chains:
-    module: "chain_injectors.hidream_smoothing_injector"
+  dynamic_hidream_o1_smoothing_chains:
+    module: "chain_injectors.hidream_o1_smoothing_injector"
+  dynamic_hidream_o1_reference_chains:
+    module: "chain_injectors.hidream_o1_reference_injector"
   
 injector_order:
   - dynamic_vae_chains
@@ -39,4 +41,5 @@ injector_order:
   - dynamic_style_chains
   - dynamic_controlnet_chains
   - dynamic_anima_controlnet_lllite_chains
-  - dynamic_hidream_smoothing_chains
+  - dynamic_hidream_o1_smoothing_chains
+  - dynamic_hidream_o1_reference_chains

yaml/model_list.yaml

@@ -92,9 +92,14 @@ Checkpoint:
           unet: "waiANIMA_v10.safetensors"
           vae: "qwen_image_vae.safetensors"
           clip: "qwen_3_06b_base.safetensors"
-      - display_name: "duongve/AnimaYume-v0.4"
+      - display_name: "duongve/AnimaYume-v0.5"
         components:
-          unet: "AnimaYume_tuned_v04.safetensors"
+          unet: "AnimaYume_tuned_v05.safetensors"
+          vae: "qwen_image_vae.safetensors"
+          clip: "qwen_3_06b_base.safetensors"
+      - display_name: "bluepen5805/Anima-pencil-v1.0"
+        components:
+          unet: "anima_pencil-v1.0.0.safetensors"
           vae: "qwen_image_vae.safetensors"
           clip: "qwen_3_06b_base.safetensors"
       - display_name: "circlestone-labs/Anima-base-v1.0"