mirror of
https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git
synced 2026-05-20 04:27:24 +08:00
BPT seed
This commit is contained in:
parent
2ff95dbd71
commit
738c302833
6
nodes.py
6
nodes.py
@ -1279,9 +1279,9 @@ class Hy3DBPT:
|
|||||||
"required": {
|
"required": {
|
||||||
"trimesh": ("TRIMESH",),
|
"trimesh": ("TRIMESH",),
|
||||||
"enable_bpt": ("BOOLEAN", {"default": True}),
|
"enable_bpt": ("BOOLEAN", {"default": True}),
|
||||||
|
"seed": ("INT", {"default": 42, "min": 0, "max": 0xffffffffffffffff}),
|
||||||
"temperature": ("FLOAT", {"default": 0.5}),
|
"temperature": ("FLOAT", {"default": 0.5}),
|
||||||
"pc_num": ("INT", {"default": 4096, "min": 1024, "max": 8192, "step": 1024}),
|
"pc_num": ("INT", {"default": 4096, "min": 1024, "max": 8192, "step": 1024}),
|
||||||
"verbose": ("BOOLEAN", {"default": False})
|
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1291,12 +1291,12 @@ class Hy3DBPT:
|
|||||||
CATEGORY = "Hunyuan3DWrapper"
|
CATEGORY = "Hunyuan3DWrapper"
|
||||||
DESCRIPTION = "BPT the mesh using bpt: https://github.com/whaohan/bpt"
|
DESCRIPTION = "BPT the mesh using bpt: https://github.com/whaohan/bpt"
|
||||||
|
|
||||||
def bpt(self, trimesh, enable_bpt, temperature, pc_num, verbose):
|
def bpt(self, trimesh, enable_bpt, temperature, pc_num, seed):
|
||||||
|
|
||||||
new_mesh = trimesh.copy()
|
new_mesh = trimesh.copy()
|
||||||
if enable_bpt:
|
if enable_bpt:
|
||||||
from .hy3dgen.shapegen.postprocessors import BptMesh
|
from .hy3dgen.shapegen.postprocessors import BptMesh
|
||||||
new_mesh = BptMesh()(new_mesh, with_normal=True, temperature=temperature, batch_size=1, pc_num=pc_num, verbose=verbose)
|
new_mesh = BptMesh()(new_mesh, with_normal=True, temperature=temperature, batch_size=1, pc_num=pc_num, verbose=False, seed=seed)
|
||||||
mm.unload_all_models()
|
mm.unload_all_models()
|
||||||
mm.soft_empty_cache()
|
mm.soft_empty_cache()
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user