--extra-index-url https://download.pytorch.org/whl/cu117

numpy==1.24.4
torch==2.0.0
torchvision==0.15.1
torchmetrics==0.6.0
pytorch-lightning==1.4.2
transformers==4.19.2
tqdm==4.66.2
gradio==3.50.2
openai==1.12.0
einops==0.3.0
-e git+https://github.com/crowsonkb/k-diffusion.git@v0.0.16#egg=k-diffusion
-e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
imageio==2.9.0
imageio-ffmpeg==0.4.2


# K_diffusion need 
accelerate
clean-fid
clip-anytorch
dctorch
einops
jsonmerge
kornia
Pillow
safetensors
scipy
torch>=2.1
torchdiffeq
torchsde
torchvision
tqdm
wandb
