ConsisID / requirements.txt
JoPmt's picture
Update requirements.txt
66a930a verified
raw
history blame
583 Bytes
--extra-index-url https://download.pytorch.org/whl/cu122
torch
torchaudio
torchvision
xformers
onnx
onnxruntime-gpu
deepspeed==0.15.2
accelerate
diffusers
transformers
tokenizers==0.20.1
peft==0.12.0
decord==0.6.0
sentencepiece==0.2.0
opencv-python==4.10.0.84
pyfacer==0.0.4
numpy==1.26.4
numba==0.60.0
insightface==0.7.3
huggingface-hub==0.26.1
facexlib==0.3.0
timm==1.0.9
func_timeout==4.3.5
tensorboard==2.17.1
gradio==4.44.1
spaces==0.29.3
pillow==10.4.0
spandrel==0.4.0
scikit-video==1.1.11
moviepy
wandb
imageio-ffmpeg
ftfy
Jinja2
einops
nvitop
optimum
omegaconf
mediapipe
tqdm