36 lines
599 B
Plaintext
36 lines
599 B
Plaintext
--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
|