update

pre-requirements.txt

@@ -2,7 +2,7 @@
 --extra-index-url https://data.dgl.ai/wheels/torch-2.1/cu121/repo.html
 torch==2.2.0
 torchvision==0.17.0
-dgl
+dgl==2.4.0
 accelerate
 beartype
 einops

primitive_anything/michelangelo/__init__.py

@@ -13,7 +13,7 @@ def load_model():
     # print(model_config)
     if hasattr(model_config, "model"):
         model_config = model_config.model
-    ckpt_path = "./ckpt/shapevae-256.ckpt"
+    ckpt_path = "./ckpt/checkpoints/aligned_shape_latents/shapevae-256.ckpt"
 
     model = instantiate_from_config(model_config, ckpt_path=ckpt_path)
     # model = model.cuda()