LoadCustomMesh

This commit is contained in:
GiusTex 2025-01-23 14:45:23 +01:00 committed by GitHub
parent 9f59f92943
commit 8fecedb105
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -3,6 +3,7 @@ import torch
from PIL import Image
from pathlib import Path
import numpy as np
import trimesh
from .hy3dgen.shapegen import Hunyuan3DDiTFlowMatchingPipeline, FaceReducer, FloaterRemover, DegenerateFaceRemover
@ -137,7 +138,29 @@ class DownloadAndLoadHy3DDelightModel:
delight_pipe.enable_model_cpu_offload()
return (delight_pipe,)
class LoadCustomMesh:
@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"}),
}
}
RETURN_TYPES = ("HY3DMESH",)
RETURN_NAMES = ("mesh",)
OUTPUT_TOOLTIPS = ("The glb model with mesh to texturize.",)
FUNCTION = "main"
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):
mesh = trimesh.load(glb, force="mesh")
return (mesh,)
class Hy3DDelightImage:
@classmethod
def INPUT_TYPES(s):