theimageconvert2 / requirements.txt
mart9992's picture
m
7423a8f
raw
history blame
1.01 kB
--extra-index-url https://download.pytorch.org/whl/cu117
torch==1.13.1+cu117
torchvision==0.14.1+cu117
torchaudio==0.13.1
accelerate
addict
gradio==3.36.1
huggingface_hub
numpy
onnxruntime
opencv_python
Pillow
pycocotools
PyYAML
requests
setuptools
supervision
termcolor
timm
gevent
yapf
numba
scipy
safetensors
pynvml
sentencepiece
loguru
# transformers@git+https://github.com/huggingface/transformers.git@main
# lama-cleaner==1.2.4
lama-cleaner@git+https://github.com/yizhangliu/lama-cleaner.git@main
mmcv==1.7.1
mmengine
openmim==0.3.9
# tiktoken
# einops
# transformers_stream_generator==0.0.4
# tensorboard
# bitsandbytes
# auto-gptq
tencentcloud-sdk-python
pydash
boto3
ninja # (Optional) Makes the build much faster
# git+https://github.com/facebookresearch/xformers.git@main#egg=xformers
git+https://github.com/huggingface/diffusers.git@main#egg=diffusers
git+https://github.com/huggingface/peft.git@main#egg=peft
git+https://github.com/huggingface/transformers.git@main#egg=transformers
matplotlib