Revert support for DA3Geometry for MoGeRender.

This commit is contained in:
Talmaj Marinc 2026-05-21 20:37:34 +02:00
parent f99c526a33
commit 976ccf43e6

View File

@ -19,9 +19,6 @@ from tqdm.auto import tqdm
MoGeModelType = io.Custom("MOGE_MODEL")
MoGeGeometry = io.Custom("MOGE_GEOMETRY")
# Redefined (not imported) to avoid a hard dependency on nodes_depth_anything_3;
# io.Custom types are matched by string key, so both definitions refer to the same wire type.
DA3Geometry = io.Custom("DA3_GEOMETRY")
# MOGE_GEOMETRY is a dict with these optional keys (absent when the upstream model didn't produce them):
# "points": torch.Tensor (B, H, W, 3)
@ -32,7 +29,6 @@ DA3Geometry = io.Custom("DA3_GEOMETRY")
# "image": torch.Tensor (B, H, W, 3) in [0, 1], CPU (always present)
def _normals_from_points(points: torch.Tensor) -> torch.Tensor:
"""Camera-space surface normals from a (B, H, W, 3) point map (v1 fallback)."""
finite = torch.isfinite(points).all(dim=-1)
@ -278,10 +274,7 @@ class MoGeRender(io.ComfyNode):
description="Render a depth map or normal map from geometry data",
category="image/geometry_estimation",
inputs=[
io.MultiType.Input("moge_geometry", types=[MoGeGeometry, DA3Geometry],
tooltip="Accepts MOGE_GEOMETRY (from MoGe nodes) or DA3_GEOMETRY (from Depth Anything 3). "
"Normal render modes require points or normals and will error if those are absent — "
"DA3 produces no point cloud, so only 'depth' and 'mask' outputs are supported."),
MoGeGeometry.Input("moge_geometry"),
io.Combo.Input("output", options=["depth", "depth_colored", "normal_opengl", "normal_directx", "mask"], default="depth",
tooltip="DirectX vs OpenGL controls the normal-map green-channel convention. DirectX: green = -Y down (Unreal). OpenGL: green = +Y up (Blender, Substance, Unity, glTF)."),
],