xinyun/ComfyUI-Hunyuan3DWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git synced 2025-12-21 10:44:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
Easymode 2025-02-19 01:13:34 +00:00 committed by GitHub
parent 3822f49560
commit 16456ebf8b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
nodes.py
View File

@ -1281,9 +1281,7 @@ class Hy3DBPT:
"enable_bpt": ("BOOLEAN", {"default": True}), "enable_bpt": ("BOOLEAN", {"default": True}),
"temperature": ("FLOAT", {"default": 0.5}), "temperature": ("FLOAT", {"default": 0.5}),
"batch_size": ("INT", {"default": 1}), "batch_size": ("INT", {"default": 1}),
"kwarg_k": ("INT", {"default": 50}), "with_normal": ("BOOLEAN", {"default": True})
"kwarg_p": ("FLOAT", {"default": 0.95}),
}, },
} }
@ -1293,11 +1291,11 @@ 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, kwarg_k, kwarg_p, temperature, batch_size): def bpt(self, trimesh, enable_bpt, with_normal, temperature, batch_size):
new_mesh = trimesh.copy() new_mesh = trimesh.copy()
if enable_bpt: if enable_bpt:
new_mesh = BptMesh()(new_mesh, max_seq_len=10000, cond_dim=768, kwarg_k=kwarg_k, kwarg_p=kwarg_p, temperature=temperature, batch_size=batch_size) new_mesh = BptMesh()(new_mesh, with_normal=with_normal, temperature=temperature, batch_size=batch_size)
mm.unload_all_models() mm.unload_all_models()
mm.soft_empty_cache() mm.soft_empty_cache()