Remove unnecessary device transfer for sound and music models in sound_generator.py

sound_generator.py

@@ -6,8 +6,8 @@ import torch
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print("Using device:", device)
 
-sound_model = AudioGen.get_pretrained('facebook/audiogen-medium').model.to(device)
-music_model = MusicGen.get_pretrained('facebook/musicgen-small').model.to(device)
+sound_model = AudioGen.get_pretrained('facebook/audiogen-medium')
+music_model = MusicGen.get_pretrained('facebook/musicgen-small')
 
 # Set generation parameters (for example, audio duration of 5 seconds)
 sound_model.set_generation_params(duration=5)