update

embodied_gen/models/sam3d.py

@@ -23,7 +23,8 @@ from typing import Optional, Union
 
 import numpy as np
 from hydra.utils import instantiate
-from modelscope import snapshot_download
+# from modelscope import snapshot_download
+from huggingface_hub import snapshot_download
 from omegaconf import OmegaConf
 from PIL import Image
 
@@ -57,7 +58,8 @@ class Sam3dInference:
         self, local_dir: str = "weights/sam-3d-objects", compile: bool = False
     ) -> None:
         if not os.path.exists(local_dir):
-            snapshot_download("facebook/sam-3d-objects", local_dir=local_dir)
+            # snapshot_download("facebook/sam-3d-objects", local_dir=local_dir)
+            snapshot_download("jetjodh/sam-3d-objects", local_dir=local_dir)
         config_file = os.path.join(local_dir, "checkpoints/pipeline.yaml")
         config = OmegaConf.load(config_file)
         config.rendering_engine = "nvdiffrast"