Add device information print statement in sound_generator.py

sound_generator.py

@@ -4,6 +4,7 @@ import torch
 
 # Load the pretrained models and move them to GPU if available
 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)