Update demo_deploy.py

demo_deploy.py

@@ -59,10 +59,10 @@ def get_model(config_file, ckpt_file, device):
     print(f"Loading policy config from {config_file}")
     policy = ACTDiffusionPolicy(policy_config)
     print(f"Loading model from {ckpt_file}")
-    policy.load_state_dict(convert_weight(torch.load(ckpt_file)["state_dict"]))
+    policy.load_state_dict(convert_weight(torch.load(ckpt_file, weights_only=False)["state_dict"]))
     policy.to(device)
     policy.eval()
-    stats = torch.load(ckpt_file)["stats"]
+    stats = torch.load(ckpt_file, weights_only=False)["stats"]
     print('Resetting observation normalization stats')
     policy.reset_obs(stats, norm_type = policy_config["norm_type"])
     camera_names = policy_config["camera_names"]