mirror of
https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git
synced 2025-12-08 20:34:28 +08:00
Updated BPT node
This commit is contained in:
parent
f2cebadfa1
commit
d34f251c36
6
nodes.py
6
nodes.py
@ -1280,6 +1280,8 @@ class Hy3DBPT:
|
||||
"trimesh": ("TRIMESH",),
|
||||
"enable_bpt": ("BOOLEAN", {"default": True}),
|
||||
"temperature": ("FLOAT", {"default": 0.5}),
|
||||
"pc_num": ("INT", {"default": 4096, "min": 1024, "max": 8192, "step": 1024}),
|
||||
"verbose": ("BOOLEAN", {"default": False})
|
||||
},
|
||||
}
|
||||
|
||||
@ -1289,12 +1291,12 @@ class Hy3DBPT:
|
||||
CATEGORY = "Hunyuan3DWrapper"
|
||||
DESCRIPTION = "BPT the mesh using bpt: https://github.com/whaohan/bpt"
|
||||
|
||||
def bpt(self, trimesh, enable_bpt, temperature):
|
||||
def bpt(self, trimesh, enable_bpt, temperature, pc_num, verbose):
|
||||
|
||||
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)
|
||||
new_mesh = BptMesh()(new_mesh, with_normal=True, temperature=temperature, batch_size=1, pc_num=pc_num, verbose=verbose)
|
||||
mm.unload_all_models()
|
||||
mm.soft_empty_cache()
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user