Option to pass in Accelerator to SoundStreamTrainer

This is needed for multi-GPU use cases where certain actions need
to be performed on the main process before the trainer is constructed.

Also minor tweak to accept more audio types in the data loader.