winnie22 / requirements.txt
mart9992's picture
m
a32aec9
accelerate
addict
diffusers
gradio==3.36.1
huggingface_hub
matplotlib
numpy
onnxruntime
opencv_python
Pillow
pycocotools
PyYAML
requests
setuptools
supervision
termcolor
timm
# torch==2.0.0 # is production
# torchvision==0.15.1 # is production
# torch
# torchvision
gevent
yapf
numba
scipy
safetensors
pynvml
sentencepiece
loguru
transformers==4.27.4
# 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