Update audio_diffusion_attacks_forhf/src/test_encoder_attack.py

audio_diffusion_attacks_forhf/src/test_encoder_attack.py

@@ -100,7 +100,8 @@ def poison_audio(waveform, sample_rate, encoders, audio_difference_weights=[1],
         music_gen_eval=XTTS_Eval(sample_rate)
     processor = AutoProcessor.from_pretrained("facebook/encodec_48khz")
     #Andy edited: loss_fn = cdpam.CDPAM(dev='cuda:0')
-    loss_fn = cdpam.CDPAM()
+    my_device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    loss_fn = cdpam.CDPAM(dev=my_device)
     for p in loss_fn.model.parameters():
         p.requires_grad = False