Update nodes.py

This commit is contained in:
kijai 2025-01-23 18:18:32 +02:00
parent ded88d8da4
commit 3b36433a87

View File

@ -139,25 +139,25 @@ class DownloadAndLoadHy3DDelightModel:
return (delight_pipe,) return (delight_pipe,)
class LoadCustomMesh: class Hy3DLoadMesh:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return { return {
"required": { "required": {
"glb": ("STRING", {"default": "", "tooltip": "The glb path with mesh to load. Tested only for now with other hunyuan3d-2 glbs"}), "glb_path": ("STRING", {"default": "", "tooltip": "The glb path with mesh to load."}),
} }
} }
RETURN_TYPES = ("HY3DMESH",) RETURN_TYPES = ("HY3DMESH",)
RETURN_NAMES = ("mesh",) RETURN_NAMES = ("mesh",)
OUTPUT_TOOLTIPS = ("The glb model with mesh to texturize.",) OUTPUT_TOOLTIPS = ("The glb model with mesh to texturize.",)
FUNCTION = "main" FUNCTION = "load"
CATEGORY = "Hunyuan3DWrapper" CATEGORY = "Hunyuan3DWrapper"
DESCRIPTION = "Encodes a text prompt using a CLIP model into an embedding that can be used to guide the diffusion model towards generating specific images." DESCRIPTION = "Encodes a text prompt using a CLIP model into an embedding that can be used to guide the diffusion model towards generating specific images."
def main(self, glb): def load(self, glb_path):
mesh = trimesh.load(glb, force="mesh") mesh = trimesh.load(glb_path, force="mesh")
return (mesh,) return (mesh,)
@ -735,7 +735,7 @@ NODE_CLASS_MAPPINGS = {
"Hy3DBakeFromMultiview": Hy3DBakeFromMultiview, "Hy3DBakeFromMultiview": Hy3DBakeFromMultiview,
"Hy3DTorchCompileSettings": Hy3DTorchCompileSettings, "Hy3DTorchCompileSettings": Hy3DTorchCompileSettings,
"Hy3DPostprocessMesh": Hy3DPostprocessMesh, "Hy3DPostprocessMesh": Hy3DPostprocessMesh,
"LoadCustomMesh": LoadCustomMesh, "Hy3DLoadMesh": Hy3DLoadMesh,
"Hy3DCameraConfig": Hy3DCameraConfig, "Hy3DCameraConfig": Hy3DCameraConfig,
"Hy3DMeshUVWrap": Hy3DMeshUVWrap, "Hy3DMeshUVWrap": Hy3DMeshUVWrap,
"Hy3DSampleMultiView": Hy3DSampleMultiView, "Hy3DSampleMultiView": Hy3DSampleMultiView,
@ -754,7 +754,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"Hy3DBakeFromMultiview": "Hy3D Bake From Multiview", "Hy3DBakeFromMultiview": "Hy3D Bake From Multiview",
"Hy3DTorchCompileSettings": "Hy3D Torch Compile Settings", "Hy3DTorchCompileSettings": "Hy3D Torch Compile Settings",
"Hy3DPostprocessMesh": "Hy3D Postprocess Mesh", "Hy3DPostprocessMesh": "Hy3D Postprocess Mesh",
"LoadCustomMesh": "Load Custom Mesh", "Hy3DLoadMesh": "Hy3D Load Mesh",
"Hy3DCameraConfig": "Hy3D Camera Config", "Hy3DCameraConfig": "Hy3D Camera Config",
"Hy3DMeshUVWrap": "Hy3D Mesh UV Wrap", "Hy3DMeshUVWrap": "Hy3D Mesh UV Wrap",
"Hy3DSampleMultiView": "Hy3D Sample MultiView", "Hy3DSampleMultiView": "Hy3D Sample MultiView",