fix load state

XPart/partgen/partformer_pipeline.py

@@ -190,7 +190,7 @@ class PartFormerPipeline(TokenAllocMixin):
                 ckpt[model_name][new_key] = value
         else:
             # ckpt = torch.load(ckpt_path, map_location="cpu", weights_only=True)
-            ckpt = torch.load(ckpt_path, map_location="cpu")
+            ckpt = torch.load(ckpt_path, weights_only=False, map_location="cpu")
         # load model
         model = instantiate_from_config(config["model"])
         # model.load_state_dict(ckpt["model"])
@@ -239,7 +239,7 @@ class PartFormerPipeline(TokenAllocMixin):
         ckpt_path = smart_load_model(
             model_path="tencent/Hunyuan3D-Part",
         )
-        ckpt = torch.load(os.path.join(ckpt_path, "xpart.pt"), map_location="cpu")
+        ckpt = torch.load(os.path.join(ckpt_path, "xpart.pt"), weights_only=False, map_location="cpu")
         # load model
         model = instantiate_from_config(config["model"])
         # model.load_state_dict(ckpt["model"])