From 241ee90164ec84a7b3d4dfd16c48e01814c790e0 Mon Sep 17 00:00:00 2001
From: Mick <mickjagger19@icloud.com>
Date: Wed, 25 Feb 2026 11:57:53 +0800
Subject: [PATCH] [diffusion] chore: tiny fix pyproject.toml (#19256)

---
 python/pyproject.toml | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

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 = [