Update audio_diffusion_attacks_forhf/src/test_encoder_attack.py

audio_diffusion_attacks_forhf/src/test_encoder_attack.py

@@ -36,7 +36,6 @@ from audiocraft.losses import (
 '''
 from audio_diffusion_attacks_forhf.src.music_gen import MusicGenEval
 from audio_diffusion_attacks_forhf.src.speech_inference import XTTS_Eval
-print("breakpoint 5")
 
 # From https://pytorch.org/tutorials/beginner/audio_preprocessing_tutorial.html#loading-audio-data-into-tensor
 def print_stats(waveform, sample_rate=None, src=None):
@@ -87,12 +86,14 @@ 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:
-        encoder.to(device='cuda')
+        #Andy removed: encoder.to(device='cuda')
         encoder.eval()
         for p in encoder.parameters():
             p.requires_grad = False
     
+    print("breakpoint 2")
     audio_len=1000000
     waveform, sample_rate = torchaudio.load(f"test_audio/Texas Sun.mp3")
     if modality=="music":