From 17f6a92a74e4533879dbdc6ac15de390d148d11a Mon Sep 17 00:00:00 2001
From: Easymode <76738305+Easymode-ai@users.noreply.github.com>
Date: Wed, 19 Feb 2025 16:02:00 +0000
Subject: [PATCH 1/3] BPT hot-fix

Reason: ImportError: cannot import name 'BptMesh' from 'ComfyUI-Hunyuan3DWrapper-main.hy3dgen.shapegen'
---
 hy3dgen/shapegen/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hy3dgen/shapegen/__init__.py b/hy3dgen/shapegen/__init__.py
index d1f9534..045564b 100755
--- a/hy3dgen/shapegen/__init__.py
+++ b/hy3dgen/shapegen/__init__.py
@@ -23,5 +23,5 @@
 # by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT.
 
 from .pipelines import Hunyuan3DDiTPipeline, Hunyuan3DDiTFlowMatchingPipeline
-from .postprocessors import FaceReducer, FloaterRemover, DegenerateFaceRemover
+from .postprocessors import FaceReducer, BptMesh, FloaterRemover, DegenerateFaceRemover
 from .preprocessors import ImageProcessorV2, IMAGE_PROCESSORS, DEFAULT_IMAGEPROCESSOR

From 1e011f62c0c54fcd193377e3edb5369a66048ceb Mon Sep 17 00:00:00 2001
From: Easymode <76738305+Easymode-ai@users.noreply.github.com>
Date: Wed, 19 Feb 2025 16:09:50 +0000
Subject: [PATCH 2/3] Revert changes

---
 hy3dgen/shapegen/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hy3dgen/shapegen/__init__.py b/hy3dgen/shapegen/__init__.py
index 045564b..d1f9534 100755
--- a/hy3dgen/shapegen/__init__.py
+++ b/hy3dgen/shapegen/__init__.py
@@ -23,5 +23,5 @@
 # by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT.
 
 from .pipelines import Hunyuan3DDiTPipeline, Hunyuan3DDiTFlowMatchingPipeline
-from .postprocessors import FaceReducer, BptMesh, FloaterRemover, DegenerateFaceRemover
+from .postprocessors import FaceReducer, FloaterRemover, DegenerateFaceRemover
 from .preprocessors import ImageProcessorV2, IMAGE_PROCESSORS, DEFAULT_IMAGEPROCESSOR

From 8c4c0f81d115a2e3dac9c2cd13373f1cf34d6715 Mon Sep 17 00:00:00 2001
From: Easymode <76738305+Easymode-ai@users.noreply.github.com>
Date: Wed, 19 Feb 2025 16:11:20 +0000
Subject: [PATCH 3/3] Fixed BPT dep error

should have been:
from .hy3dgen.shapegen.postprocessors import BptMesh
---
 nodes.py | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/nodes.py b/nodes.py
index e57f1ff..b829502 100644
--- a/nodes.py
+++ b/nodes.py
@@ -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: