diff --git a/demo.ipynb b/demo.ipynb
index 8c4f2ca..4f2e10d 100644
--- a/demo.ipynb
+++ b/demo.ipynb
@@ -112,7 +112,7 @@
     "\n",
     "# Install necessary packages\n",
     "!pip install torch torchvision torchaudio\n",
-    "!pip install transformers accelerate peft huggingface_hub decord av opencv-python matplotlib einops jaxtyping num2words"
+    "!pip install transformers accelerate peft huggingface_hub decord av opencv-python matplotlib einops jaxtyping num2words \"torchao>=0.16.0\"\n"
    ]
   },
   {