got a response here https://github.com/lucidrains/soundstorm-pytorch/discussions...