diff --git a/nodes.py b/nodes.py index 3768969..5602beb 100644 --- a/nodes.py +++ b/nodes.py @@ -1282,6 +1282,7 @@ class Hy3DBPT: "seed": ("INT", {"default": 42, "min": 0, "max": 0xffffffffffffffff}), "temperature": ("FLOAT", {"default": 0.5}), "pc_num": ("INT", {"default": 4096, "min": 1024, "max": 8192, "step": 1024}), + "samples": ("INT", {"default": 50000}) }, } @@ -1291,12 +1292,12 @@ class Hy3DBPT: CATEGORY = "Hunyuan3DWrapper" DESCRIPTION = "BPT the mesh using bpt: https://github.com/whaohan/bpt" - def bpt(self, trimesh, enable_bpt, temperature, pc_num, seed): + def bpt(self, trimesh, enable_bpt, temperature, pc_num, seed, samples): new_mesh = trimesh.copy() if enable_bpt: from .hy3dgen.shapegen.postprocessors import BptMesh - new_mesh = BptMesh()(new_mesh, with_normal=True, temperature=temperature, batch_size=1, pc_num=pc_num, verbose=False, seed=seed) + new_mesh = BptMesh()(new_mesh, with_normal=True, temperature=temperature, batch_size=1, pc_num=pc_num, verbose=False, seed=seed, samples=samples) mm.unload_all_models() mm.soft_empty_cache()