From cc742005d74f56811330da437fc692275d5e0a89 Mon Sep 17 00:00:00 2001
From: Easymode <76738305+Easymode-ai@users.noreply.github.com>
Date: Thu, 20 Feb 2025 02:29:01 +0000
Subject: [PATCH] Update nodes.py

Clear memory before starting BPT operation
---
 nodes.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index 5958c62..9d1a52a 100644
--- a/nodes.py
+++ b/nodes.py
@@ -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