diff --git a/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/comfy_ui_dependencies.rs b/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/comfy_ui_dependencies.rs
index 5e80cd9c99..4b90d1da6f 100644
--- a/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/comfy_ui_dependencies.rs
+++ b/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/comfy_ui_dependencies.rs
@@ -155,6 +155,18 @@ impl RequiredModels {
                     location: PathBuf::from("models/grounding-dino/groundingdino_swint_ogc.pth"),
                     url: "https://storage.googleapis.com/vocodes-public/comfyui-deps/groundingdino_swint_ogc.pth".to_string(),
                 },
+                ComfyDependency {
+                    location: PathBuf::from("models/grounding-dino/groundingdino_swint_ogc.pth"),
+                    url: "https://storage.googleapis.com/vocodes-public/comfyui-deps/groundingdino_swint_ogc.pth".to_string(),
+                },
+                ComfyDependency {
+                    location: PathBuf::from("models/grounding-dino/groundingdino_swint_ogc.pth"),
+                    url: "https://storage.googleapis.com/vocodes-public/comfyui-deps/groundingdino_swint_ogc.pth".to_string(),
+                },
+                ComfyDependency {
+                    location: PathBuf::from("models/vae/vae-ft-mse-840000-ema-pruned.ckpt"),
+                    url: "https://storage.googleapis.com/vocodes-public/comfyui-deps/vae-ft-mse-840000-ema-pruned.ckpt".to_string(),
+                },        
             ],
         }
     }