VAE / requirements.txt
souranil3d's picture
First commit for VAE space
16906c1
torch==1.8.0
python-box==5.3.0
tensorboardX==2.1
pydantic
streamlit==0.82.0
streamlit-drawable-canvas==0.8.0
pytorch_lightning>=1.1.1
torchvision==0.9.0