Text-to-video example outputs noisy video in AWS

I was working with the text-to-video example here. https://twitter.com/multimodalart/status/1638617810866675740

It works fine in Colab but outputs garbage video when I run in my AWS EC2/SageMaker notebook instance. The !nvidia-smi output from my SageMaker instance is below if that helps