FacePoke / requirements.txt
jbilcke-hf's picture
jbilcke-hf HF staff
improve compatibility by using onnx 1.16.1
ee15487
# --------------------------------------------------------------------
# Cuda 12.4
# --------------------------------------------------------------------
--extra-index-url https://download.pytorch.org/whl/cu124
torch
torchvision
torchaudio
torchgeometry
# --------------------------------------------------------------------
# Common libraries for LivePortrait and all
# --------------------------------------------------------------------
# LRU cache compatible with asyncio
async-lru==2.0.4
# note: gradio is only used for the cropping utility
gradio==5.0.0b4
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 has some issues it seems:
# https://github.com/onnx/onnx/issues/6267
#onnx==1.16.2
# update: I've rolled it back to 1.16.1
# see: https://github.com/jbilcke-hf/FacePoke/issues/23#issuecomment-2414714490
onnx==1.16.1
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
transformers==4.39.2
gdown==5.2.0
requests==2.32.3
omegaconf==2.3.0
pydantic==2.9.2
# --------------------------------------------------------------------
# RESERVED FOR FUTURE USAGE
#
# (it adds bloat, so you can remove them if you want)
# --------------------------------------------------------------------
aiohttp==3.10.5
av==12.3.0
einops==0.7.0
safetensors==0.4.5
huggingface-hub==0.25.1
optimum-quanto==0.2.4
# --------------------------------------------------------------------
# Used for advanced LivePortrait features
# --------------------------------------------------------------------
pillow==10.4.0
# by popular demand, let's add AVIF
pillow-avif-plugin==1.4.6