torch | |
torchaudio | |
torchsde | |
torchvision | |
einops | |
transformers | |
tokenizers>=0.13.3 | |
sentencepiece | |
safetensors>=0.4.2 | |
aiohttp | |
Pillow | |
tqdm | |
psutil | |
#non essential dependencies: | |
kornia>=0.7.1 | |
spandrel | |
soundfile | |
huggingface-hub | |
accelerate | |
color-matcher | |
matplotlib | |
cachetools | |
colorama | |
importlib_metadata | |
scipy | |
opencv-python==4.7.0.72 | |
scikit-image | |
pyyaml | |
python-dateutil | |
mediapipe | |
svglib | |
fvcore | |
yapf | |
omegaconf | |
ftfy | |
addict | |
yacs | |
trimesh[easy] | |
albumentations | |
scikit-learn | |
git+https://github.com/WASasquatch/img2texture.git | |
git+https://github.com/WASasquatch/cstr | |
imageio | |
joblib | |
numba | |
pilgram | |
timm>=0.4.12 | |
piexif | |
loguru | |
python-decouple |