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.
Loading
Please register or sign in to comment