Upload folder using huggingface_hub

core/pipelines/workflow_recipes/_partials/conditioning/pixeldit.yaml

@@ -0,0 +1,43 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Diffusion Model"
+    params:
+      weight_dtype: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load VAE"
+    params:
+      vae_name: "pixel_space"
+  clip_loader:
+    class_type: CLIPLoader
+    title: "Load CLIP"
+    params:
+      type: "pixeldit"
+      device: "default"
+
+connections:
+  - from: "unet_loader:0"
+    to: "ksampler:model"
+  - from: "clip_loader:0"
+    to: "pos_prompt:clip"
+  - from: "clip_loader:0"
+    to: "neg_prompt:clip"
+  - from: "pos_prompt:0"
+    to: "ksampler:positive"
+  - from: "neg_prompt:0"
+    to: "ksampler:negative"
+  - from: "vae_loader:0"
+    to: "vae_decode:vae"
+  - from: "vae_loader:0"
+    to: "vae_encode:vae"
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "clip_loader:0"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  vae_name: "vae_loader:vae_name"
+  clip_name: "clip_loader:clip_name"

ui/shared/hires_fix_ui.py

@@ -52,11 +52,11 @@ def create_ui():
                     components[f'denoise_{prefix}'] = gr.Slider(label="Denoise Strength", minimum=0.0, maximum=1.0, step=0.01, value=0.55)
                 
                 with gr.Row():
-                    components[f'sampler_{prefix}'] = gr.Dropdown(label="Sampler", choices=SAMPLER_CHOICES, value=SAMPLER_CHOICES[0])
-                    components[f'scheduler_{prefix}'] = gr.Dropdown(label="Scheduler", choices=SCHEDULER_CHOICES, value='normal' if 'normal' in SCHEDULER_CHOICES else SCHEDULER_CHOICES[0])
+                    components[f'sampler_{prefix}'] = gr.Dropdown(label="Sampler", choices=SAMPLER_CHOICES, value='er_sde' if 'er_sde' in SAMPLER_CHOICES else SAMPLER_CHOICES[0])
+                    components[f'scheduler_{prefix}'] = gr.Dropdown(label="Scheduler", choices=SCHEDULER_CHOICES, value='simple' if 'simple' in SCHEDULER_CHOICES else SCHEDULER_CHOICES[0])
                 with gr.Row():
-                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=20)
-                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=5.0)
+                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=30)
+                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=4.0)
                 with gr.Row():
                     components[f'seed_{prefix}'] = gr.Number(label="Seed (-1 for random)", value=-1, precision=0)
                     components[f'batch_size_{prefix}'] = gr.Slider(label="Batch Size", minimum=1, maximum=16, step=1, value=1)

ui/shared/img2img_ui.py

@@ -36,11 +36,11 @@ def create_ui():
                 components[f'denoise_{prefix}'] = gr.Slider(label="Denoise Strength", minimum=0.0, maximum=1.0, step=0.01, value=0.7)
                 
                 with gr.Row():
-                    components[f'sampler_{prefix}'] = gr.Dropdown(label="Sampler", choices=SAMPLER_CHOICES, value=SAMPLER_CHOICES[0])
-                    components[f'scheduler_{prefix}'] = gr.Dropdown(label="Scheduler", choices=SCHEDULER_CHOICES, value='normal' if 'normal' in SCHEDULER_CHOICES else SCHEDULER_CHOICES[0])
+                    components[f'sampler_{prefix}'] = gr.Dropdown(label="Sampler", choices=SAMPLER_CHOICES, value='er_sde' if 'er_sde' in SAMPLER_CHOICES else SAMPLER_CHOICES[0])
+                    components[f'scheduler_{prefix}'] = gr.Dropdown(label="Scheduler", choices=SCHEDULER_CHOICES, value='simple' if 'simple' in SCHEDULER_CHOICES else SCHEDULER_CHOICES[0])
                 with gr.Row():
-                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=20)
-                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=5.0)
+                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=30)
+                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=4.0)
                 with gr.Row():
                     components[f'seed_{prefix}'] = gr.Number(label="Seed (-1 for random)", value=-1, precision=0)
                     components[f'batch_size_{prefix}'] = gr.Slider(label="Batch Size", minimum=1, maximum=16, step=1, value=1)

ui/shared/inpaint_ui.py

@@ -62,11 +62,11 @@ def create_ui():
                         label="Grow Mask By", minimum=0, maximum=64, step=1, value=6
                     )
                 with gr.Row():
-                    components[f'sampler_{prefix}'] = gr.Dropdown(label="Sampler", choices=SAMPLER_CHOICES, value=SAMPLER_CHOICES[0])
-                    components[f'scheduler_{prefix}'] = gr.Dropdown(label="Scheduler", choices=SCHEDULER_CHOICES, value='normal' if 'normal' in SCHEDULER_CHOICES else SCHEDULER_CHOICES[0])
+                    components[f'sampler_{prefix}'] = gr.Dropdown(label="Sampler", choices=SAMPLER_CHOICES, value='er_sde' if 'er_sde' in SAMPLER_CHOICES else SAMPLER_CHOICES[0])
+                    components[f'scheduler_{prefix}'] = gr.Dropdown(label="Scheduler", choices=SCHEDULER_CHOICES, value='simple' if 'simple' in SCHEDULER_CHOICES else SCHEDULER_CHOICES[0])
                 with gr.Row():
-                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=20)
-                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=5.0)
+                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=30)
+                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=4.0)
                 with gr.Row():
                     components[f'seed_{prefix}'] = gr.Number(label="Seed (-1 for random)", value=-1, precision=0)
                     components[f'batch_size_{prefix}'] = gr.Slider(label="Batch Size", minimum=1, maximum=16, step=1, value=1)

ui/shared/outpaint_ui.py

@@ -48,11 +48,11 @@ def create_ui():
                 components[f'feathering_{prefix}'] = gr.Slider(label="Feathering / Grow Mask", minimum=0, maximum=100, step=1, value=10)
                 
                 with gr.Row():
-                    components[f'sampler_{prefix}'] = gr.Dropdown(label="Sampler", choices=SAMPLER_CHOICES, value=SAMPLER_CHOICES[0])
-                    components[f'scheduler_{prefix}'] = gr.Dropdown(label="Scheduler", choices=SCHEDULER_CHOICES, value='normal' if 'normal' in SCHEDULER_CHOICES else SCHEDULER_CHOICES[0])
+                    components[f'sampler_{prefix}'] = gr.Dropdown(label="Sampler", choices=SAMPLER_CHOICES, value='er_sde' if 'er_sde' in SAMPLER_CHOICES else SAMPLER_CHOICES[0])
+                    components[f'scheduler_{prefix}'] = gr.Dropdown(label="Scheduler", choices=SCHEDULER_CHOICES, value='simple' if 'simple' in SCHEDULER_CHOICES else SCHEDULER_CHOICES[0])
                 with gr.Row():
-                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=20)
-                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=5.0)
+                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=30)
+                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=4.0)
                 with gr.Row():
                     components[f'seed_{prefix}'] = gr.Number(label="Seed (-1 for random)", value=-1, precision=0)
                     components[f'batch_size_{prefix}'] = gr.Slider(label="Batch Size", minimum=1, maximum=16, step=1, value=1)

ui/shared/ui_components.py

@@ -90,11 +90,11 @@ def create_base_parameter_ui(prefix, defaults=None):
         components[f'width_{prefix}'] = gr.Number(label="Width", value=defaults.get('w', 1024), interactive=True)
         components[f'height_{prefix}'] = gr.Number(label="Height", value=defaults.get('h', 1024), interactive=True)
     with gr.Row():
-        components[f'sampler_{prefix}'] = gr.Dropdown(label="Sampler", choices=SAMPLER_CHOICES, value=SAMPLER_CHOICES[0])
-        components[f'scheduler_{prefix}'] = gr.Dropdown(label="Scheduler", choices=SCHEDULER_CHOICES, value='normal' if 'normal' in SCHEDULER_CHOICES else SCHEDULER_CHOICES[0])
+        components[f'sampler_{prefix}'] = gr.Dropdown(label="Sampler", choices=SAMPLER_CHOICES, value='er_sde' if 'er_sde' in SAMPLER_CHOICES else SAMPLER_CHOICES[0])
+        components[f'scheduler_{prefix}'] = gr.Dropdown(label="Scheduler", choices=SCHEDULER_CHOICES, value='simple' if 'simple' in SCHEDULER_CHOICES else SCHEDULER_CHOICES[0])
     with gr.Row():
-        components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=20)
-        components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=5.0)
+                    components[f'steps_{prefix}'] = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=30)
+                    components[f'cfg_{prefix}'] = gr.Slider(label="CFG Scale", minimum=1.0, maximum=20.0, step=0.1, value=4.0)
     with gr.Row():
         components[f'seed_{prefix}'] = gr.Number(label="Seed (-1 for random)", value=-1, precision=0)
         components[f'batch_size_{prefix}'] = gr.Slider(label="Batch Size", minimum=1, maximum=16, step=1, value=1)

yaml/constants.yaml

@@ -7,6 +7,22 @@ MAX_REFERENCE_LATENTS: 10
 LORA_SOURCE_CHOICES: ["Civitai", "File"]
 
 RESOLUTION_MAP:
+  pixeldit:
+    "1:1 (Square)": [1024, 1024]
+    "16:9 (Landscape)": [1344, 768]
+    "9:16 (Portrait)": [768, 1344]
+    "4:3 (Classic)": [1152, 896]
+    "3:4 (Classic Portrait)": [896, 1152]
+    "3:2 (Photography)": [1216, 832]
+    "2:3 (Photography Portrait)": [832, 1216]
+  lens:
+    "1:1 (Square)": [1024, 1024]
+    "16:9 (Landscape)": [1344, 768]
+    "9:16 (Portrait)": [768, 1344]
+    "4:3 (Classic)": [1152, 896]
+    "3:4 (Classic Portrait)": [896, 1152]
+    "3:2 (Photography)": [1216, 832]
+    "2:3 (Photography Portrait)": [832, 1216]
   ernie-image:
     "1:1 (Square)": [1024, 1024]
     "16:9 (Landscape)": [1344, 768]

yaml/file_list.yaml

@@ -403,6 +403,11 @@ file:
       repo_id: "comfyanonymous/ControlNet-v1-1_fp16_safetensors"
       repository_file_path: "control_v11u_sd15_tile_fp16.safetensors"
   diffusion_models: 
+    # PixelDiT
+    - filename: "pixeldit_1300m_1024px_mxfp8.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/PixelDiT"
+      repository_file_path: "diffusion_models/pixeldit_1300m_1024px_mxfp8.safetensors"
     # Lens
     - filename: "lens_mxfp8.safetensors"
       source: "hf"
@@ -682,6 +687,11 @@ file:
       repo_id: "alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union-2.1"
       repository_file_path: "Z-Image-Turbo-Fun-Controlnet-Tile-2.1-8steps.safetensors"
   text_encoders: 
+    # PixelDiT
+    - filename: "gemma_2_2b_it_elm_fp8_scaled.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/PixelDiT"
+      repository_file_path: "text_encoders/gemma_2_2b_it_elm_fp8_scaled.safetensors"
     # Lens
     - filename: "gpt_oss_20b_nvfp4.safetensors"
       source: "hf"

yaml/image_gen_features.yaml

@@ -7,6 +7,10 @@ default:
     - style
     - conditioning
 
+pixeldit:
+  enabled_chains:
+  - conditioning
+
 lens:
   enabled_chains:
   - conditioning

yaml/model_architectures.yaml

@@ -1,4 +1,5 @@
 architecture_order:
+  - "PixelDiT"
   - "Lens"
   - "FLUX.2-KV"
   - "FLUX.2"
@@ -22,6 +23,9 @@ architecture_order:
   - "SD1.5"
 
 architectures:
+  "PixelDiT":
+    model_type: "pixeldit"
+    controlnet_key: "PixelDiT"
   "Lens":
     model_type: "lens"
     controlnet_key: "Lens"

yaml/model_defaults.yaml

@@ -6,13 +6,19 @@ Default:
   positive_prompt: ""
   negative_prompt: ""
 
+PixelDiT:
+  _defaults:
+    steps: 30
+    cfg: 4.0
+    sampler_name: "er_sde"
+    scheduler: "simple"
+
 Lens:
   _defaults:
     steps: 20
     cfg: 5.0
     sampler_name: "euler"
     scheduler: "simple"
-    total_pixels: 1048576
   "Microsoft/Lens-Turbo":
     steps: 4
     cfg: 1.0

yaml/model_list.yaml

@@ -1,4 +1,12 @@
 Checkpoint:
+  PixelDiT:
+    latent_type: chroma_radiance_latent
+    models:
+      - display_name: "nvidia/PixelDiT-1300M-1024px"
+        components:
+          unet: "pixeldit_1300m_1024px_mxfp8.safetensors"
+          clip: "gemma_2_2b_it_elm_fp8_scaled.safetensors"
+          vae: "pixel_space"
   Lens:
     latent_type: flux2_latent
     models: