# --------------------------------------------------------------------
# Cuda 12.4
# --------------------------------------------------------------------

--extra-index-url https://download.pytorch.org/whl/cu124
torch
torchvision
torchaudio


# --------------------------------------------------------------------
# Common libraries for LivePortrait and all
# --------------------------------------------------------------------

# note: gradio is only used for the cropping utility
gradio==5.0.0b4

gradio-webrtc==0.0.1

opuslib==3.0.1

pyyaml==6.0.1
numpy==1.22.0 # Updated to resolve conflicts
opencv-python==4.8.1.78 # Downgraded to be compatible with numpy
scipy==1.10.1
imageio==2.31.1
imageio-ffmpeg==0.5.1
lmdb==1.4.1
tqdm==4.66.4
rich==13.7.1

# this one is from 5 years ago, we should probably
# use python-ffmpeg or typed-ffmpeg instead
# https://pypi.org/search/?q=ffmpeg-python
ffmpeg-python==0.2.0

# versions of onnx gpu <= 1.17 did not support Cuda 12
onnxruntime-gpu==1.19.2
onnx==1.16.2
scikit-image==0.20.0
albumentations==1.3.1
matplotlib==3.7.2
tyro==0.8.5
chumpy==0.70

diffusers==0.30.3
accelerate==0.34.2
tensorflow==2.12.0
tensorboard==2.12.0
soundfile==0.12.1
transformers==4.39.2

gdown==5.2.0
requests==2.32.3
omegaconf==2.3.0
moviepy==1.0.3

pydantic==2.9.2

# --------------------------------------------------------------------
# RESERVED FOR FUTURE USAGE
#
# (it adds bloat, so you can remove them if you want)
# --------------------------------------------------------------------
dnspython>=2.0.0
ifaddr>=0.2.0
aioice==0.9.0
aiortc==1.9.0
aiohttp==3.10.5
av==12.3.0
einops==0.7.0
safetensors==0.4.5
sentencepiece==0.2.0
sounddevice==0.5.0
sphn==0.1.4
huggingface-hub==0.25.1
optimum-quanto==0.2.4
httpx==0.24.1
timm==1.0.9
ftfy==6.2.3
facexlib==0.3.0

# --------------------------------------------------------------------
# Used for advanced LivePortrait features
# --------------------------------------------------------------------
pillow==10.4.0