# PyTorch 2.4.1 + torchvision
torch==2.4.1
torchvision==0.19.1
torchaudio==2.4.1
# Core dependencies
diffusers
transformers>=4.34.0
accelerate
pyrallis
easydict
omegaconf==2.1.1
einops==0.3.0
torch-fidelity==0.3.0
torchmetrics>=0.11.0
pytorch-lightning==2.0.9
taming-transformers @ git+https://github.com/CompVis/taming-transformers.git@master
clip @ git+https://github.com/openai/CLIP.git@main

# Computer vision
opencv-python==4.5.5.64              # ✅ 避免 dnn.DictValue 错误
opencv-python-headless==4.5.5.64     # ✅ 防止系统默认装新版导致冲突（和 GUI 冲突不大）
kornia==0.6
albumentations==0.4.3
imageio>=2.27                        # ✅ 保证兼容 scikit-image 0.21.0
imageio-ffmpeg==0.4.2
matplotlib
scikit-image==0.21.0
Pillow
segment-anything
numpy==1.24.4                        # ✅ 避免 numpy 2.x 引发 ABI 问题              

# Gradio
gradio
gradio-bbox-annotator

# Utilities
natsort
roifile
fill-voids
configargparse
ipywidgets
ftfy
sniffio
websocket-client
dask