From 3b36433a87a667e74ca3928d1ba8ab225f350703 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Thu, 23 Jan 2025 18:18:32 +0200
Subject: [PATCH] Update nodes.py

---
 nodes.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/nodes.py b/nodes.py
index fff1be1..197d6be 100644
--- a/nodes.py
+++ b/nodes.py
@@ -139,25 +139,25 @@ class DownloadAndLoadHy3DDelightModel:
         
         return (delight_pipe,)
 
-class LoadCustomMesh:
+class Hy3DLoadMesh:
     @classmethod
     def INPUT_TYPES(s):
         return {
             "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_NAMES = ("mesh",)
     OUTPUT_TOOLTIPS = ("The glb model with mesh to texturize.",)
     
-    FUNCTION = "main"
+    FUNCTION = "load"
     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."
 
-    def main(self, glb):
+    def load(self, glb_path):
         
-        mesh = trimesh.load(glb, force="mesh")
+        mesh = trimesh.load(glb_path, force="mesh")
         
         return (mesh,)
         
@@ -735,7 +735,7 @@ NODE_CLASS_MAPPINGS = {
     "Hy3DBakeFromMultiview": Hy3DBakeFromMultiview,
     "Hy3DTorchCompileSettings": Hy3DTorchCompileSettings,
     "Hy3DPostprocessMesh": Hy3DPostprocessMesh,
-    "LoadCustomMesh": LoadCustomMesh,
+    "Hy3DLoadMesh": Hy3DLoadMesh,
     "Hy3DCameraConfig": Hy3DCameraConfig,
     "Hy3DMeshUVWrap": Hy3DMeshUVWrap,
     "Hy3DSampleMultiView": Hy3DSampleMultiView,
@@ -754,7 +754,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "Hy3DBakeFromMultiview": "Hy3D Bake From Multiview",
     "Hy3DTorchCompileSettings": "Hy3D Torch Compile Settings",
     "Hy3DPostprocessMesh": "Hy3D Postprocess Mesh",
-    "LoadCustomMesh": "Load Custom Mesh",
+    "Hy3DLoadMesh": "Hy3D Load Mesh",
     "Hy3DCameraConfig": "Hy3D Camera Config",
     "Hy3DMeshUVWrap": "Hy3D Mesh UV Wrap",
     "Hy3DSampleMultiView": "Hy3D Sample MultiView",