From d34f251c362a469a086232e748056e8c3f4df753 Mon Sep 17 00:00:00 2001
From: Easymode <76738305+Easymode-ai@users.noreply.github.com>
Date: Wed, 19 Feb 2025 17:16:35 +0000
Subject: [PATCH] Updated BPT node

---
 nodes.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index 429c5c9..047c990 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1280,6 +1280,8 @@ class Hy3DBPT:
                 "trimesh": ("TRIMESH",),
                 "enable_bpt": ("BOOLEAN", {"default": True}),
                 "temperature": ("FLOAT", {"default": 0.5}),
+                "pc_num": ("INT", {"default": 4096, "min": 1024, "max": 8192, "step": 1024}),
+                "verbose": ("BOOLEAN", {"default": False})
             },
         }
 
@@ -1289,12 +1291,12 @@ class Hy3DBPT:
     CATEGORY = "Hunyuan3DWrapper"
     DESCRIPTION = "BPT the mesh using bpt: https://github.com/whaohan/bpt"
     
-    def bpt(self, trimesh, enable_bpt, temperature):
+    def bpt(self, trimesh, enable_bpt, temperature, pc_num, verbose):
 
         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=1)
+            new_mesh = BptMesh()(new_mesh, with_normal=True, temperature=temperature, batch_size=1, pc_num=pc_num, verbose=verbose)
             mm.unload_all_models()
             mm.soft_empty_cache()