Fix model loading in sound_generator.py to access the model attribute directly

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').to(device)
-music_model = MusicGen.get_pretrained('facebook/musicgen-small').to(device)
+sound_model = AudioGen.get_pretrained('facebook/audiogen-medium').model.to(device)
+music_model = MusicGen.get_pretrained('facebook/musicgen-small').model.to(device)
 
 # Set generation parameters (for example, audio duration of 5 seconds)
 sound_model.set_generation_params(duration=5)