Update trellis/pipelines/trellis_image_to_3d.py

trellis/pipelines/trellis_image_to_3d.py

@@ -19,7 +19,7 @@ from typing import *
 from scipy.spatial.transform import Rotation
 from transformers import AutoModelForImageSegmentation
 import rembg
-from dreamsim import dreamsim
+# from dreamsim import dreamsim
 
 def export_point_cloud(xyz, color):
     # Convert tensors to numpy arrays if needed
@@ -987,8 +987,8 @@ 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()
+        # 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