sjc / requirements.txt
amankishore's picture
taming
4e3f695
pydantic
tqdm
click
easydict
tabulate
imageio
einops
matplotlib
omegaconf==2.1.1
torchmetrics==0.6.0
pytorch-lightning==1.4.2
transformers
kornia==0.6.0
git+https:///github.com/openai/CLIP.git#egg=clip
imageio[ffmpeg]
imageio[pyav]
--extra-index-url https://download.pytorch.org/whl/cu116
torch
torchvision
torchaudio
-e git+https://github.com/CompVis/taming-transformers.git#egg=taming-transformers