Fix torch.load: add weights_only=False for PyTorch 2.6 compat

app.py

@@ -49,7 +49,7 @@ def generate_audio(video_file, seed_val, cfg_scale, num_steps, mode):
     from samplers import euler_sampler, euler_maruyama_sampler
     from diffusers import AudioLDM2Pipeline
     extract_cavp = Extract_CAVP_Features(device=device, config_path="./cavp/cavp.yaml", ckpt_path=cavp_ckpt_path)
-    state_dict = torch.load(onset_ckpt_path, map_location=device)["state_dict"]
+    state_dict = torch.load(onset_ckpt_path, map_location=device, weights_only=False)["state_dict"]
     new_state_dict = {}
     for key, value in state_dict.items():
         if "model.net.model" in key:
@@ -63,7 +63,7 @@ def generate_audio(video_file, seed_val, cfg_scale, num_steps, mode):
     onset_model.load_state_dict(new_state_dict)
     onset_model.eval()
     model = MMDiT(adm_in_channels=120, z_dims=[768], encoder_depth=4).to(device)
-    ckpt = torch.load(taro_ckpt_path, map_location=device)["ema"]
+    ckpt = torch.load(taro_ckpt_path, map_location=device, weights_only=False)["ema"]
     model.load_state_dict(ckpt)
     model.eval()
     model.to(weight_dtype)