diff --git a/python/pyproject.toml b/python/pyproject.toml
index 4704af72f..e7065ad49 100755
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -19,7 +19,6 @@ dependencies = [
   "aiohttp",
   "apache-tvm-ffi>=0.1.5,<0.2",
   "anthropic>=0.20.0",
-  "av",
   "blobfile==3.0.0",
   "build",
   "compressed-tensors",
@@ -102,18 +101,19 @@ torch = [
 checkpoint-engine = ["checkpoint-engine==0.1.2"]
 diffusion = [
   "PyYAML==6.0.1",
-  "cloudpickle",
+  "cloudpickle==3.1.2",
   "diffusers==0.36.0",
   "imageio==2.36.0",
   "imageio-ffmpeg==0.5.1",
   "moviepy>=2.0.0",
   "opencv-python-headless==4.10.0.84",
-  "remote-pdb",
+  "remote-pdb==2.1.0",
   "st_attn==0.0.7 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
   "vsa==0.0.4 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
   "runai_model_streamer>=0.15.5",
   "cache-dit==1.2.0",
-  "addict"
+  "addict==2.4.0",
+  "av==16.1.0",
 ]
 
 tracing = [