Update trellis/pipelines/trellis_image_to_3d.py

trellis/pipelines/trellis_image_to_3d.py

@@ -19,8 +19,9 @@ from typing import *
 from scipy.spatial.transform import Rotation
 from transformers import AutoModelForImageSegmentation
 import rembg
-from dreamsim import dreamsim
-from tqdm import tqdm
+# for app_refine.py, please uncomment these lines
+# from dreamsim import dreamsim 
+# from tqdm import tqdm
 
 def export_point_cloud(xyz, color):
     # Convert tensors to numpy arrays if needed
@@ -1062,8 +1063,9 @@ class TrellisVGGTTo3DPipeline(TrellisImageTo3DPipeline):
 
         new_pipeline._init_image_cond_model(args['image_cond_model'])
 
-        model, _ = dreamsim(pretrained=True, device=new_pipeline.device, dreamsim_type="dino_vitb16", cache_dir="weights/dreamsim")
-        new_pipeline.dreamsim_model = model
-        new_pipeline.dreamsim_model.eval()
+        # for app_refine.py, please uncomment these lines
+        # model, _ = dreamsim(pretrained=True, device=new_pipeline.device, dreamsim_type="dino_vitb16", cache_dir="weights/dreamsim")
+        # new_pipeline.dreamsim_model = model
+        # new_pipeline.dreamsim_model.eval()
 
         return new_pipeline