xinyun/ComfyUI-Hunyuan3DWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git synced 2026-05-18 15:47:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
Easymode 2025-02-19 00:47:51 +00:00 committed by GitHub
parent 439e2e01d2
commit 74ede84b22
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nodes.py
View File

@ -1279,10 +1279,11 @@ class Hy3DBPT:
"required": { "required": {
"trimesh": ("TRIMESH",), "trimesh": ("TRIMESH",),
"enable_bpt": ("BOOLEAN", {"default": True}), "enable_bpt": ("BOOLEAN", {"default": True}),
"cond_dim": ("INT", {"default": 768}), "temperature": ("FLOAT", {"default": 0.5}),
"max_seq_len": ("INT", {"default": 10000}), "batch_size": ("INT", {"default": 1}),
"kwarg_k": ("INT", {"default": 50}), "kwarg_k": ("INT", {"default": 50}),
"kwarg_p": ("FLOAT", {"default": 0.95}), "kwarg_p": ("FLOAT", {"default": 0.95}),
}, },
} }
@ -1292,11 +1293,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, cond_dim, max_seq_len, kwarg_k, kwarg_p): def bpt(self, trimesh, enable_bpt, kwarg_k, kwarg_p, 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=max_seq_len, cond_dim=cond_dim, kwarg_k=kwarg_k, kwarg_p=kwarg_p) 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)
mm.unload_all_models() mm.unload_all_models()
mm.soft_empty_cache() mm.soft_empty_cache()