Update nodes.py

Clear memory before starting BPT operation
This commit is contained in:
Easymode 2025-02-20 02:29:01 +00:00 committed by GitHub
parent c7e0ab75e6
commit cc742005d7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1293,7 +1293,8 @@ class Hy3DBPT:
DESCRIPTION = "BPT the mesh using bpt: https://github.com/whaohan/bpt"
def bpt(self, trimesh, enable_bpt, temperature, pc_num, seed, samples):
mm.unload_all_models()
mm.soft_empty_cache()
new_mesh = trimesh.copy()
if enable_bpt:
from .hy3dgen.shapegen.postprocessors import BptMesh