Upload folder using huggingface_hub

README.md

@@ -40,6 +40,7 @@ models:
   - Comfy-Org/HiDream-I1_ComfyUI
   - Comfy-Org/HiDream-O1-Image
   - Comfy-Org/HunyuanImage_2.1_ComfyUI
+  - Comfy-Org/Lens
   - Comfy-Org/LongCat-Image
   - Comfy-Org/Lumina_Image_2.0_Repackaged
   - Comfy-Org/NewBie-image-Exp0.1_repackaged

app.py

@@ -3,6 +3,10 @@ import os
 import sys
 import site
 
+if "--use-sage-attention" not in sys.argv:
+    sys.argv.append("--use-sage-attention")
+    print("🚀 [SageAttention] Injected '--use-sage-attention' into sys.argv.")
+
 APP_DIR = os.path.dirname(os.path.abspath(__file__))
 if APP_DIR not in sys.path:
     sys.path.insert(0, APP_DIR)
@@ -52,6 +56,10 @@ def main():
     print("--- [Setup] Starting ComfyUI initialization ---")
     setup_comfyui.initialize_comfyui()
     
+    print("--- [Setup] Applying SageAttention Runtime Patch ---")
+    patch_result = apply_sage_attention_patch()
+    print(f"--- [Setup] {patch_result} ---")
+
     print("--- [Setup] Reloading site-packages to detect newly installed packages... ---")
     try:
         site.main()
@@ -59,7 +67,7 @@ def main():
     except Exception as e:
         print(f"--- [Setup] ⚠️  Warning: Could not fully reload site-packages: {e} ---")
 
-    print("--- Initiating GPU Startup Check & SageAttention Patch ---")
+    print("--- Initiating GPU Startup Check & SageAttention Patch Verification ---")
     try:
         dummy_gpu_for_startup()
     except Exception as e:

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

@@ -0,0 +1,54 @@
+nodes:
+  unet_loader:
+    class_type: UNETLoader
+    title: "Load Diffusion Model"
+    params:
+      weight_dtype: "default"
+  clip_loader:
+    class_type: CLIPLoader
+    title: "Load CLIP"
+    params:
+      type: "lens"
+      device: "default"
+  vae_loader:
+    class_type: VAELoader
+    title: "Load VAE"
+
+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_lora_chains:
+  lora_chain:
+    template: "LoraLoader"
+    output_map:
+      "unet_loader:0": "model"
+      "clip_loader:0": "clip"
+    input_map:
+      "model": "model"
+      "clip": "clip"
+    end_input_map:
+      "model": ["ksampler:model"]
+      "clip": ["pos_prompt:clip", "neg_prompt:clip"]
+
+dynamic_conditioning_chains:
+  conditioning_chain:
+    ksampler_node: "ksampler"
+    clip_source: "clip_loader:0"
+
+ui_map:
+  unet_name: "unet_loader:unet_name"
+  clip_name: "clip_loader:clip_name"
+  vae_name: "vae_loader:vae_name"

requirements.txt

@@ -41,6 +41,6 @@ diffusers
 protobuf
 huggingface-hub
 imageio
-
+ninja
 spaces
 sageattention @ https://huggingface.co/RioShiina/Sage-Attention-ZeroGPU-Space-Build/resolve/main/sageattention-2.2.0-cp312-cp312-linux_x86_64.whl

yaml/file_list.yaml

@@ -403,6 +403,15 @@ file:
       repo_id: "comfyanonymous/ControlNet-v1-1_fp16_safetensors"
       repository_file_path: "control_v11u_sd15_tile_fp16.safetensors"
   diffusion_models: 
+    # Lens
+    - filename: "lens_mxfp8.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/Lens"
+      repository_file_path: "diffusion_models/lens_mxfp8.safetensors"
+    - filename: "lens_turbo_mxfp8.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/Lens"
+      repository_file_path: "diffusion_models/lens_turbo_mxfp8.safetensors"
     # Anima
     - filename: "waiANIMA_v10.safetensors"
       source: "hf"
@@ -673,6 +682,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: 
+    # Lens
+    - filename: "gpt_oss_20b_nvfp4.safetensors"
+      source: "hf"
+      repo_id: "Comfy-Org/Lens"
+      repository_file_path: "text_encoders/gpt_oss_20b_nvfp4.safetensors"
     # Anima
     - filename: "qwen_3_06b_base.safetensors"
       source: "hf"

yaml/image_gen_features.yaml

@@ -7,6 +7,10 @@ default:
     - style
     - conditioning
 
+lens:
+  enabled_chains:
+  - conditioning
+
 ernie-image:
   enabled_chains:
     - lora

yaml/model_architectures.yaml

@@ -1,4 +1,5 @@
 architecture_order:
+  - "Lens"
   - "FLUX.2-KV"
   - "FLUX.2"
   - "ERNIE-Image"
@@ -21,6 +22,9 @@ architecture_order:
   - "SD1.5"
 
 architectures:
+  "Lens":
+    model_type: "lens"
+    controlnet_key: "Lens"
   "ERNIE-Image":
     model_type: "ernie-image"
     controlnet_key: "ERNIE-Image"

yaml/model_defaults.yaml

@@ -6,6 +6,17 @@ Default:
   positive_prompt: ""
   negative_prompt: ""
 
+Lens:
+  _defaults:
+    steps: 20
+    cfg: 5.0
+    sampler_name: "euler"
+    scheduler: "simple"
+    total_pixels: 1048576
+  "Microsoft/Lens-Turbo":
+    steps: 4
+    cfg: 1.0
+
 ERNIE-Image:
   _defaults:
     steps: 20

yaml/model_list.yaml

@@ -1,4 +1,17 @@
 Checkpoint:
+  Lens:
+    latent_type: flux2_latent
+    models:
+      - display_name: "Microsoft/Lens"
+        components:
+          unet: "lens_mxfp8.safetensors"
+          clip: "gpt_oss_20b_nvfp4.safetensors"
+          vae: "flux2-vae.safetensors"
+      - display_name: "Microsoft/Lens-Turbo"
+        components:
+          unet: "lens_turbo_mxfp8.safetensors"
+          clip: "gpt_oss_20b_nvfp4.safetensors"
+          vae: "flux2-vae.safetensors"
   FLUX.2-KV:
     latent_type: flux2_latent
     models: