VITRA / requirements.txt
arnoldland's picture
Initial commit
aae3ba1
torch==2.4.0
torchvision==0.19.0
torchaudio==2.4.0
PyYAML==6.0.2
hydra-colorlog>=1.2.0
hydra-core>=1.1.1
deepspeed==0.16.5
tensorboard>=2.13.0
tensorboardX>=2.6.2
tqdm>=4.65.0
transformers==4.47.1
diffusers>=0.31.0
wandb>=0.19.0
numpy<2.0
sentence-transformers==2.2.2
open_clip_torch==2.20.0
datasets==2.12.0
draccus==0.8.0
einops
huggingface_hub
json-numpy
jsonlines
matplotlib
peft==0.11.1
protobuf
rich
sentencepiece==0.1.99
timm==0.9.10
tokenizers>=0.21
decord
scikit-image
brotli
imageio-ffmpeg
imageio
ffmpeg-python
opencv-python
pympler
ultralytics
pytorch-lightning
yacs
gradio
git+https://github.com/EasternJournalist/utils3d.git@d790d33#egg=utils3d
smplx
projectaria_tools
https://github.com/arnoldland/chumpy/releases/download/v0.71/chumpy-0.71-py3-none-any.whl
https://github.com/arnoldland/pytorch3d/releases/download/v0.2/pytorch3d-0.7.8-cp310-cp310-linux_x86_64.whl