BF16 version of : https://huggingface.co./HKUST-Audio/xcodec2/tree/main

Use `with torch.autocast(device_type='cuda')` when doing inference