torch==1.13.1+cu117
torchvision==0.14.1+cu117
torchaudio==0.13.1
tyro
--extra-index-url https://download.pytorch.org/whl/cu117
pytorch-lightning
numpy
tqdm
PyYAML
git+https://github.com/openai/CLIP.git
einops
huggingface-hub
imageio
imageio-ffmpeg
importlib-resources
ipdb
ipython
jedi
kiwisolver
kornia
lpips
matplotlib
matplotlib-inline
omegaconf
open-clip-torch
opencv-python
parso
pathtools
pexpect
pickleshare
pillow
prompt-toolkit
protobuf
psutil
ptyprocess
pure-eval
pygments
pymcubes
pyparsing
pytorch-fid
pytorch-msssim
regex
safetensors
scipy
sentencepiece
sentry-sdk
setproctitle
smmap
stack-data
timm
tokenizers
tomli
traitlets
transformers
trimesh
vit-pytorch
wandb
wcwidth
zipp
kiui
accelerate