PIA / environment.yaml
LeoXing1996
init repo for fg
a001281
raw
history blame contribute delete
376 Bytes
name: pia
channels:
- pytorch
- nvidia
dependencies:
- python=3.10
- pytorch=1.13.1
- torchvision=0.14.1
- torchaudio=0.13.1
- pytorch-cuda=11.7
- pip
- pip:
- diffusers==0.24.0
- transformers==4.25.1
- xformers==0.0.16
- imageio==2.27.0
- decord==0.6.0
- gdown
- einops
- omegaconf
- safetensors
- gradio
- wandb