diff --git a/src/platform/assets/mappings/modelNodeMappings.ts b/src/platform/assets/mappings/modelNodeMappings.ts index 9ce9da8667..e1c3e20992 100644 --- a/src/platform/assets/mappings/modelNodeMappings.ts +++ b/src/platform/assets/mappings/modelNodeMappings.ts @@ -210,5 +210,11 @@ export const MODEL_NODE_MAPPINGS: ReadonlyArray< // ---- Ultralytics YOLO detectors (ComfyUI-Impact-Pack) ---- ['ultralytics/bbox', 'UltralyticsDetectorProvider', 'model_name'], - ['ultralytics/segm', 'UltralyticsDetectorProvider', 'model_name'] + ['ultralytics/segm', 'UltralyticsDetectorProvider', 'model_name'], + + // ---- ComfyUI core geometry estimation (MoGe) ---- + ['geometry_estimation', 'LoadMoGeModel', 'model_name'], + + // ---- ComfyUI core optical flow (RAFT) ---- + ['optical_flow', 'OpticalFlowLoader', 'model_name'] ] as const satisfies ReadonlyArray