xinyun/ComfyUI-Hunyuan3DWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git synced 2026-01-24 00:14:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

BPT samples added to node

Browse Source
This commit is contained in:
Easymode 2025-02-19 18:54:03 +00:00 committed by GitHub
parent 3b543e82bc
commit 74cfd26eda
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nodes.py
View File

@ -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()