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

Remove BPT batch size as it's per node

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

5
nodes.py
View File

@ -1280,7 +1280,6 @@ class Hy3DBPT:
"trimesh": ("TRIMESH",),
"enable_bpt": ("BOOLEAN", {"default": True}),
"temperature": ("FLOAT", {"default": 0.5}),
"batch_size": ("INT", {"default": 1}),
},
}
@ -1290,12 +1289,12 @@ class Hy3DBPT:
CATEGORY = "Hunyuan3DWrapper"
DESCRIPTION = "BPT the mesh using bpt: https://github.com/whaohan/bpt"
def bpt(self, trimesh, enable_bpt, temperature, batch_size):
def bpt(self, trimesh, enable_bpt, temperature):
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=batch_size)
new_mesh = BptMesh()(new_mesh, with_normal=True, temperature=temperature, batch_size=1)
mm.unload_all_models()
mm.soft_empty_cache()