Update freesplatter/webui/runner.py

freesplatter/webui/runner.py

@@ -2,6 +2,7 @@ import os
 import json
 import uuid
 import time
+import shutil
 import rembg
 import numpy as np
 import trimesh
@@ -83,6 +84,17 @@ class FreeSplatterRunner:
         # self.rembg = rembg.new_session('birefnet-general')
 
         # diffusion models
+        download_dir = snapshot_download('tencent/Hunyuan3D-1', repo_type='model')
+        shutil.rmtree(os.path.join(download_dir, 'assets'))
+        shutil.rmtree(os.path.join(download_dir, 'mvd_lite'))
+        shutil.rmtree(os.path.join(download_dir, 'svrm'))
+        pipeline = HunYuan3D_MVD_Std_Pipeline.from_pretrained(
+            os.path.join(download_dir, 'mvd_std'),
+            torch_dtype=torch.float16,
+            use_safetensors=True,
+        )
+        self.hunyuan3d_mvd_std = pipeline.to(device)
+
         pipeline = DiffusionPipeline.from_pretrained(
             "sudo-ai/zero123plus-v1.1", 
             custom_pipeline="sudo-ai/zero123plus-pipeline",
@@ -103,14 +115,6 @@ class FreeSplatterRunner:
         )
         self.zero123plus_v12 = pipeline.to(device)
 
-        download_dir = snapshot_download('tencent/Hunyuan3D-1', repo_type='model')
-        pipeline = HunYuan3D_MVD_Std_Pipeline.from_pretrained(
-            os.path.join(download_dir, 'mvd_std'),
-            torch_dtype=torch.float16,
-            use_safetensors=True,
-        )
-        self.hunyuan3d_mvd_std = pipeline.to(device)
-
         # freesplatter
         config_file = 'configs/freesplatter-object.yaml'
         ckpt_path = hf_hub_download('TencentARC/FreeSplatter', repo_type='model', filename='freesplatter-object.safetensors')