Update audio_diffusion_attacks_forhf/src/music_gen.py

audio_diffusion_attacks_forhf/src/music_gen.py

@@ -73,9 +73,9 @@ class MusicGenEval:
 
 model_name="facebook/musicgen-stereo-small"
 processor = AutoProcessor.from_pretrained(model_name)
-model = MusicgenForConditionalGeneration.from_pretrained(model_name).to(device='cuda')
+#Andy commented (hesitant): model = MusicgenForConditionalGeneration.from_pretrained(model_name).to(device='cuda')
 
-'''Andy commented:
+'''Andy commented (hesitant):
 song_name="Texas Sun"
 waveform, sample_rate = torchaudio.load(f"test_audio/{song_name}.mp3")
 waveform=waveform[:, :500000]