The_Broccolator / requirements.txt
plaidam's picture
Update requirements.txt
38d9ce5 verified
# Core dependencies
numpy==1.25.0
scipy>=1.11.0
pillow>=9.5.0
tqdm
psutil
einops
torchsde
spandrel
kornia>=0.7.1
gguf
replicate
numba
# PyTorch and AI ecosystem (handled separately or pinned in your Docker/base environment)
# torch==2.4.0
# torchvision
torchaudio
accelerate
diffusers
transformers
xformers
fairscale>=0.4.4
timm>=0.4.12
pytorch-lightning
blend_modes
surrealist
# CUDA-dependent packages (installed after CUDA setup)
#inplace-abn==1.1.0
# WAS Node Suite dependencies
git+https://github.com/WASasquatch/img2texture.git
git+https://github.com/WASasquatch/cstr
git+https://github.com/WASasquatch/ffmpy.git
rembg
pilgram
joblib
imageio
# Dependencies needed for detectron2 & DensePose, but NOT detectron2 itself
# (We install detectron2 & DensePose explicitly in app.py instead)
future
fvcore>=0.1.5
iopath>=0.1.7
omegaconf>=2.1
hydra-core>=1.1
black==21.4b2
yacs>=0.1.8
tabulate
pycocotools>=2.0.2
termcolor>=1.1
boto3==1.34.86
redis
fal-client
# Removed lines:
# git+https://github.com/facebookresearch/[email protected]
# git+https://github.com/facebookresearch/[email protected]#subdirectory=projects/DensePose