Update difpoint/inference.py

difpoint/inference.py

@@ -159,7 +159,7 @@ class Inferencer(object):
         self.device = 'cuda'
         from difpoint.model import get_model
         self.point_diffusion = get_model()
-        ckpt = torch.load('./downloaded_repo/ckpts/KDTalker.pth', weights_only=True)
+        ckpt = torch.load('./downloaded_repo/ckpts/KDTalker.pth', weights_only=False)
 
         self.point_diffusion.load_state_dict(ckpt['model'])
         print('model', self.point_diffusion.children())