diff --git a/comfyui/workflows/text-to-image/flux-dev-t2i-production-v1.json b/comfyui/workflows/text-to-image/flux-dev-t2i-production-v1.json
index d272e9d..c8ff980 100644
--- a/comfyui/workflows/text-to-image/flux-dev-t2i-production-v1.json
+++ b/comfyui/workflows/text-to-image/flux-dev-t2i-production-v1.json
@@ -192,8 +192,8 @@
       "properties": {
         "Node name for S&R": "VAELoader"
       },
-      "widgets_values": ["pixel_space"],
-      "title": "VAE Loader"
+      "widgets_values": ["ae.safetensors"],
+      "title": "FLUX VAE Loader"
     },
     {
       "id": 7,
diff --git a/comfyui_models.yaml b/comfyui_models.yaml
index a8315dc..1fda906 100644
--- a/comfyui_models.yaml
+++ b/comfyui_models.yaml
@@ -277,6 +277,19 @@ model_categories:
         - source: "text_encoders/t5xxl_fp16.safetensors"
           dest: "t5xxl_fp16.safetensors"
 
+    - repo_id: black-forest-labs/FLUX.1-schnell
+      description: FLUX VAE - Autoencoder for FLUX models
+      size_gb: 0.5
+      essential: true
+      category: support
+      type: vae
+      format: safetensors
+      vram_gb: 1
+      notes: VAE autoencoder required for FLUX image decoding
+      files:
+        - source: "ae.safetensors"
+          dest: "ae.safetensors"
+
     - repo_id: stabilityai/stable-diffusion-3.5-large
       description: T5-XXL FP16 - For CogVideoX text encoding
       size_gb: 9