wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-14 04:22:31 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Make Load3D model_file optional by adding "none" choice

Browse Source
This commit is contained in:
Terry Jia 2026-04-12 20:59:55 -04:00
parent 31283d2892
commit 4fe06a265c
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
comfy_extras/nodes_load_3d.py
View File

@ -34,7 +34,7 @@ class Load3D(IO.ComfyNode):
essentials_category="Basics", essentials_category="Basics",
is_experimental=True, is_experimental=True,
inputs=[ inputs=[
IO.Combo.Input("model_file", options=sorted(files), upload=IO.UploadType.model), IO.Combo.Input("model_file", options=["none"] + sorted(files), upload=IO.UploadType.model),
IO.Load3D.Input("image"), IO.Load3D.Input("image"),
IO.Int.Input("width", default=1024, min=1, max=4096, step=1), IO.Int.Input("width", default=1024, min=1, max=4096, step=1),
IO.Int.Input("height", default=1024, min=1, max=4096, step=1), IO.Int.Input("height", default=1024, min=1, max=4096, step=1),
@ -68,8 +68,12 @@ class Load3D(IO.ComfyNode):
video = InputImpl.VideoFromFile(recording_video_path) video = InputImpl.VideoFromFile(recording_video_path)
file_3d = Types.File3D(folder_paths.get_annotated_filepath(model_file)) file_3d = None
return IO.NodeOutput(output_image, output_mask, model_file, normal_image, image['camera_info'], video, file_3d) mesh_path = ""
if model_file and model_file != "none":
file_3d = Types.File3D(folder_paths.get_annotated_filepath(model_file))
mesh_path = model_file
return IO.NodeOutput(output_image, output_mask, mesh_path, normal_image, image['camera_info'], video, file_3d)
process = execute # TODO: remove process = execute # TODO: remove