# Core dependencies
torch
torchvision
torchaudio
transformers==4.33.1
accelerate==0.22.0
gradio>=4.0.0
spaces

# Model dependencies
diffusers @ git+https://github.com/TMElyralab/diffusers.git@tme
ip_adapter @ git+https://github.com/tencent-ailab/IP-Adapter.git@main
clip @ git+https://github.com/openai/CLIP.git@main
controlnet_aux @ git+https://github.com/TMElyralab/controlnet_aux.git@tme
mmcm @ git+https://github.com/TMElyralab/MMCM.git@setup

# Additional dependencies
ninja==1.11.1
bitsandbytes==0.41.1
xformers==0.0.21
omegaconf
einops
imageio==2.31.1
pandas
matplotlib
pynvml==11.5.0
moviepy==1.0.3
scikit-learn
librosa==0.10.1
ffmpeg-python
easydict
mediapipe==0.10.3
insightface==0.7.3
opencv-python
Pillow
protobuf==3.20.3
PyYAML
scipy
tqdm
albumentations==1.3.1
opencv-contrib-python==4.8.0.76
imageio-ffmpeg==0.4.8
pytorch-lightning==2.0.8
timm==0.9.12
safetensors==0.3.3
pycocotools
kornia==0.7.0
open_clip_torch==2.20.0
torchmetrics==1.1.1
invisible-watermark==0.1.5
ftfy==6.1.1
loguru==0.6.0
markupsafe==2.0.1
cuid