S3Diff / environment.yaml
zhangap's picture
Upload 213 files
36d9761 verified
raw
history blame
691 Bytes
name: s3diff
channels:
- pytorch
- defaults
dependencies:
- python=3.10
- pip:
- einops>=0.6.1
- numpy>=1.24.4
- open-clip-torch>=2.20.0
- opencv-python==4.6.0.66
- pillow>=9.5.0
- scipy==1.11.1
- timm>=0.9.2
- tokenizers
- torch>=2.0.1
- torchaudio>=2.0.2
- torchdata==0.6.1
- torchmetrics>=1.0.1
- torchvision>=0.15.2
- tqdm>=4.65.0
- transformers==4.35.2
- triton==2.0.0
- urllib3<1.27,>=1.25.4
- xformers>=0.0.20
- streamlit-keyup==0.2.0
- lpips
- peft
- pyiqa
- omegaconf
- dominate
- diffusers==0.25.1
- gradio==3.43.1