requirements.txt

# ===== Core DL stack (your versions, CPU/ZeroGPU friendly) =====
torch==2.2.0
torchvision==0.17.0
torchaudio==2.2.0
# 若你的 Space 使用带 CUDA 12.1 的 GPU，可以把上面三行替换为下面三行：
# torch==2.2.0+cu121 --extra-index-url https://download.pytorch.org/whl/cu121
# torchvision==0.17.0+cu121 --extra-index-url https://download.pytorch.org/whl/cu121
# torchaudio==2.2.0+cu121 --extra-index-url https://download.pytorch.org/whl/cu121

# xformers 与 torch 2.2.0 的兼容版本
xformers==0.0.24

# ===== Numerics & utils =====
pandas==2.2.2
einops==0.4.1
easydict==1.13
tqdm==4.66.4

# ===== Image/Video IO =====
opencv-python-headless==4.10.0.84
Pillow==10.4.0
matplotlib==3.8.4
imageio==2.19.3
imageio-ffmpeg==0.4.7
decord==0.6.0
OpenEXR==3.3.1
sentencepiece
peft>=0.12.0
# ===== HF ecosystem =====
transformers==4.48.3
safetensors>=0.4.2
huggingface-hub>=0.24.0
accelerate>=0.33.0
numpy==1.26.4
# 你仓库里带有本地修改的 diffusers（推荐保留）
# -e code_edit/diffusers
# 如果不放本地 diffusers，请删掉上一行，改用：
# diffusers>=0.31.0

# ===== App & logging =====
gradio>=4.44.0
spaces>=0.42.0
wandb>=0.17.0