Merge pull request #69 from Easymode-ai/main

BPT hot-fix
This commit is contained in:
Jukka Seppänen 2025-02-19 18:13:51 +02:00 committed by GitHub
commit e3b3fb25e7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1281,7 +1281,6 @@ class Hy3DBPT:
"enable_bpt": ("BOOLEAN", {"default": True}),
"temperature": ("FLOAT", {"default": 0.5}),
"batch_size": ("INT", {"default": 1}),
"with_normal": ("BOOLEAN", {"default": True})
},
}
@ -1290,16 +1289,16 @@ class Hy3DBPT:
FUNCTION = "bpt"
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, with_normal, temperature, batch_size):
from .hy3dgen.shapegen import BptMesh
new_mesh = trimesh.copy()
if enable_bpt:
new_mesh = BptMesh()(new_mesh, with_normal=with_normal, temperature=temperature, batch_size=batch_size)
mm.unload_all_models()
mm.soft_empty_cache()
from .hy3dgen.shapegen.postprocessors import BptMesh
new_mesh = BptMesh()(new_mesh, with_normal=True, temperature=temperature, batch_size=batch_size)
mm.unload_all_models()
mm.soft_empty_cache()
return (new_mesh, )
class Hy3DGetMeshPBRTextures: