Update requirements.txt

requirements.txt

@@ -1,34 +1,44 @@
+# Torch suite strictly pinned to match the flash-attn wheel below
 torch==2.6.0
 torchvision==0.21.0
 torchdata==0.10.1
 
+# Pre-compiled Flash Attention (must match Python 3.10 and Torch 2.6)
+flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
+
+# Gradio strictly pinned to perfectly match your README.md sdk_version
+gradio==5.33.0
+
+# Core AI and processing libraries (loosened to >= to prevent pip resolution errors)
 opencv-python>=4.9.0.80
 diffusers>=0.31.0
-transformers==4.51.3
-#transformers==4.46.3 # was needed by llamallava used by i2v hunyuan before patch
+transformers>=4.51.3
 tokenizers>=0.20.3
 accelerate>=1.1.1
+peft>=0.17.0
+mmgp>=3.4.8
+rembg[gpu]>=2.0.65
+
+# Prevent Numpy 2.x breaking changes
+numpy>=1.23.5,<2
+
+# Prevent MoviePy 2.x breaking changes (v2 completely changed the API)
+moviepy==1.0.3
+
+# Pydantic loosened to allow Gradio to use its preferred version
+pydantic>=2.10.6
+
+# Unpinned Image/Video/Audio utilities (pip will automatically find the best versions)
 tqdm
 imageio
 easydict
 ftfy
 dashscope
 imageio-ffmpeg
-
-# flash_attn
-flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
-
-gradio==5.33.0   
-numpy>=1.23.5,<2
 einops
-moviepy==1.0.3
-mmgp==3.4.8
-peft>=0.17.0
 mutagen
-pydantic>=2.10.6
 decord
 onnxruntime-gpu
-rembg[gpu]==2.0.65
 matplotlib
 timm
 segment-anything
@@ -38,4 +48,3 @@ librosa
 loguru
 sentencepiece
 av
-# rembg==2.0.65