The_Broccolator / requirements.txt
plaidam's picture
Update requirements.txt
6b1a58c verified
raw
history blame
958 Bytes
# 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 (installed via install.sh)
# torch==2.4.0
# torchvision
# torchaudio
accelerate
diffusers
transformers
xformers
fairscale>=0.4.4
timm>=0.4.12
pytorch-lightning
# 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
# detectron2 and densepose
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
git+https://github.com/facebookresearch/[email protected] # Base detectron2
git+https://github.com/facebookresearch/[email protected]#subdirectory=projects/DensePose
# Rest of your dependencies...