From c899ce1a4de450b35d3ac9174fdf5cdb0a213935 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 31 Jan 2025 00:44:42 +0200
Subject: [PATCH] Update nodes.py

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

diff --git a/nodes.py b/nodes.py
index cd82e55..e0a6c7b 100644
--- a/nodes.py
+++ b/nodes.py
@@ -576,6 +576,9 @@ class Hy3DRenderMultiViewDepth:
 
     def process(self, mesh, render_size, texture_size, camera_config=None):
 
+        mm.unload_all_models()
+        mm.soft_empty_cache()
+
         from .hy3dgen.texgen.differentiable_renderer.mesh_render import MeshRender
 
         if camera_config is None:
@@ -970,6 +973,9 @@ class Hy3DGenerateMesh:
 
     def process(self, pipeline, image, steps, guidance_scale, seed, mask=None):
 
+        mm.unload_all_models()
+        mm.soft_empty_cache()
+
         device = mm.get_torch_device()
         offload_device = mm.unet_offload_device()