update

app.py

@@ -11,6 +11,7 @@ from torchvision import transforms
 from huggingface_hub import hf_hub_download, snapshot_download
 import subprocess
 import shutil
+import gc
 
 # install others
 #subprocess.run("pip install spandrel==0.4.1 --no-deps", shell=True, check=True)
@@ -159,6 +160,7 @@ def run_full(image: str, req: gr.Request):
     mesh.export(mesh_path)
     print("save to ", mesh_path)
 
+    gc.collect()
     torch.cuda.empty_cache()
 
     height, width = 768, 768
@@ -221,6 +223,7 @@ def run_full(image: str, req: gr.Request):
         **pipe_kwargs,
     ).images
 
+    gc.collect()
     torch.cuda.empty_cache()
 
     mv_image_path = os.path.join(save_dir, f"mv_adapter_{get_random_hex()}.png")
@@ -244,6 +247,9 @@ def run_full(image: str, req: gr.Request):
         camera_azimuth_deg=[x - 90 for x in [0, 90, 180, 270, 180, 180]],
     )
 
+    gc.collect()
+    torch.cuda.empty_cache()
+
     return image_seg, mesh_path, textured_glb_path
     
 
@@ -283,6 +289,7 @@ def image_to_3d(
     mesh.export(mesh_path)
     print("save to ", mesh_path)
 
+    gc.collect()
     torch.cuda.empty_cache()
 
     return mesh_path
@@ -350,6 +357,7 @@ def run_texture(image: Image, mesh_path: str, seed: int, req: gr.Request):
         **pipe_kwargs,
     ).images
 
+    gc.collect()
     torch.cuda.empty_cache()
 
     save_dir = os.path.join(TMP_DIR, str(req.session_hash))
@@ -373,6 +381,8 @@ def run_texture(image: Image, mesh_path: str, seed: int, req: gr.Request):
         rgb_process_config=ModProcessConfig(view_upscale=True, inpaint_mode="view"),
         camera_azimuth_deg=[x - 90 for x in [0, 90, 180, 270, 180, 180]],
     )
+    gc.collect()
+    torch.cuda.empty_cache()
 
     return textured_glb_path