Update audio_diffusion_attacks_forhf/src/test_encoder_attack.py

audio_diffusion_attacks_forhf/src/test_encoder_attack.py

@@ -86,14 +86,12 @@ def poison_audio(waveform, sample_rate, encoders, audio_difference_weights=[1],
     audio_folder: string, path to folder of audio files. Protected audio files will be saved in that folder.
     encoders: encoders to protect against. See initialization at end of file.
     '''
-    print("breakpoint 1")
     for encoder in encoders:
         #Andy removed: encoder.to(device='cuda')
         encoder.eval()
         for p in encoder.parameters():
             p.requires_grad = False
     
-    print("breakpoint 2")
     audio_len=1000000
     #Andy removed: waveform, sample_rate = torchaudio.load(f"test_audio/Texas Sun.mp3")
     if modality=="music":
@@ -101,7 +99,8 @@ def poison_audio(waveform, sample_rate, encoders, audio_difference_weights=[1],
     elif modality=="speech":
         music_gen_eval=XTTS_Eval(sample_rate)
     processor = AutoProcessor.from_pretrained("facebook/encodec_48khz")
-    loss_fn = cdpam.CDPAM(dev='cuda:0')
+    #Andy edited: loss_fn = cdpam.CDPAM(dev='cuda:0')
+    loss_fn = cdpam.CDPAM()
     for p in loss_fn.model.parameters():
         p.requires_grad = False