mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-24 16:07:30 +08:00
Fixup last few naming issues.
This commit is contained in:
parent
976ccf43e6
commit
2c3be9dd66
@ -157,7 +157,7 @@ class DepthAnything3Inference(io.ComfyNode):
|
|||||||
category="image/geometry_estimation",
|
category="image/geometry_estimation",
|
||||||
description="Run Depth Anything 3 on an image or image batch. In multi-view mode each frame is treated as a separate view of the same scene.",
|
description="Run Depth Anything 3 on an image or image batch. In multi-view mode each frame is treated as a separate view of the same scene.",
|
||||||
inputs=[
|
inputs=[
|
||||||
io.Model.Input("model"),
|
DA3ModelType.Input("da3_model"),
|
||||||
io.Image.Input("image",
|
io.Image.Input("image",
|
||||||
tooltip="Single image or image batch. "
|
tooltip="Single image or image batch. "
|
||||||
"In multi-view mode each frame is treated as "
|
"In multi-view mode each frame is treated as "
|
||||||
@ -219,14 +219,14 @@ class DepthAnything3Inference(io.ComfyNode):
|
|||||||
)
|
)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def execute(cls, model, image, process_res, resize_method, mode) -> io.NodeOutput:
|
def execute(cls, da3_model, image, process_res, resize_method, mode) -> io.NodeOutput:
|
||||||
mode_val = mode["mode"] # "mono" or "multiview"
|
mode_val = mode["mode"] # "mono" or "multiview"
|
||||||
|
|
||||||
if mode_val == "mono":
|
if mode_val == "mono":
|
||||||
return cls._execute_mono(model, image, process_res, resize_method)
|
return cls._execute_mono(da3_model, image, process_res, resize_method)
|
||||||
|
|
||||||
# Capability checks for multi-view pose.
|
# Capability checks for multi-view pose.
|
||||||
diffusion = model.model.diffusion_model
|
diffusion = da3_model.model.diffusion_model
|
||||||
pose_method = mode["pose_method"]
|
pose_method = mode["pose_method"]
|
||||||
ref_view_strategy = mode["ref_view_strategy"]
|
ref_view_strategy = mode["ref_view_strategy"]
|
||||||
|
|
||||||
@ -244,7 +244,7 @@ class DepthAnything3Inference(io.ComfyNode):
|
|||||||
)
|
)
|
||||||
|
|
||||||
return cls._execute_multiview(
|
return cls._execute_multiview(
|
||||||
model, image, process_res, resize_method,
|
da3_model, image, process_res, resize_method,
|
||||||
ref_view_strategy, pose_method,
|
ref_view_strategy, pose_method,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user