wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-27 01:17:24 +08:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

Make LoadDepthAnything3 look for models in geometry_estimation.

Browse Source
This commit is contained in:
Talmaj Marinc 2026-05-19 20:25:09 +02:00
parent 5b1490315b
commit 893ab85a8f
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy_extras/nodes_depth_anything_3.py
View File

@ -36,6 +36,7 @@ import folder_paths
from comfy.ldm.depth_anything_3 import preprocess as da3_preprocess from comfy.ldm.depth_anything_3 import preprocess as da3_preprocess
from comfy_api.latest import ComfyExtension, io from comfy_api.latest import ComfyExtension, io
class LoadDepthAnything3(io.ComfyNode): class LoadDepthAnything3(io.ComfyNode):
@classmethod @classmethod
def define_schema(cls): def define_schema(cls):
@ -67,7 +68,7 @@ class LoadDepthAnything3(io.ComfyNode):
elif weight_dtype == "fp32": elif weight_dtype == "fp32":
model_options["dtype"] = torch.float32 model_options["dtype"] = torch.float32
path = folder_paths.get_full_path_or_raise("depth_estimation", model_name) path = folder_paths.get_full_path_or_raise("geometry_estimation", model_name)
model = comfy.sd.load_diffusion_model(path, model_options=model_options) model = comfy.sd.load_diffusion_model(path, model_options=model_options)
return io.NodeOutput(model) return io.NodeOutput(model)