Skip to content

Commit f4004f6

Browse files
authored
TOIL(requirements): Split requirements to per-platform (#355)
* toil(reqs): split requirements to per-platform Signed-off-by: Ben Alkov <[email protected]> * toil(reqs): fix for Win and Lin... ...allow pip to resolve latest torch, numpy Signed-off-by: Ben Alkov <[email protected]> * toil(install): update reqs in Win install notebook Signed-off-by: Ben Alkov <[email protected]> Signed-off-by: Ben Alkov <[email protected]>
1 parent 4406fd1 commit f4004f6

File tree

4 files changed

+75
-5
lines changed

4 files changed

+75
-5
lines changed

Stable-Diffusion-local-Windows.ipynb

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -65,25 +65,31 @@
6565
"imageio-ffmpeg==0.4.2\n",
6666
"imageio==2.9.0\n",
6767
"kornia==0.6.0\n",
68+
"# pip will resolve the version which matches torch\n",
69+
"numpy\n",
6870
"omegaconf==2.1.1\n",
6971
"opencv-python==4.6.0.66\n",
7072
"pillow==9.2.0\n",
73+
"pip>=22\n",
7174
"pudb==2019.2\n",
7275
"pytorch-lightning==1.4.2\n",
7376
"streamlit==1.12.0\n",
74-
"# Regular \"taming-transformers\" doesn't seem to work\n",
77+
"# \"CompVis/taming-transformers\" doesn't work\n",
78+
"# ldm\\models\\autoencoder.py\", line 6, in <module>\n",
79+
"# from taming.modules.vqvae.quantize import VectorQuantizer2 as VectorQuantizer\n",
80+
"# ModuleNotFoundError\n",
7581
"taming-transformers-rom1504==0.0.6\n",
7682
"test-tube>=0.7.5\n",
7783
"torch-fidelity==0.3.0\n",
7884
"torchmetrics==0.6.0\n",
79-
"torchvision==0.12.0\n",
8085
"transformers==4.19.2\n",
8186
"git+https://github.com/openai/CLIP.git@main#egg=clip\n",
8287
"git+https://github.com/lstein/k-diffusion.git@master#egg=k-diffusion\n",
8388
"# No CUDA in PyPi builds\n",
84-
"torch@https://download.pytorch.org/whl/cu113/torch-1.11.0%2Bcu113-cp310-cp310-win_amd64.whl\n",
85-
"# No MKL in PyPi builds (faster, more robust than OpenBLAS)\n",
86-
"numpy@https://download.lfd.uci.edu/pythonlibs/archived/numpy-1.22.4+mkl-cp310-cp310-win_amd64.whl\n",
89+
"--extra-index-url https://download.pytorch.org/whl/cu113 --trusted-host https://download.pytorch.org\n",
90+
"torch==1.11.0\n",
91+
"# Same as numpy - let pip do its thing\n",
92+
"torchvision\n",
8793
"-e .\n"
8894
]
8995
},

requirements-lin.txt

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
albumentations==0.4.3
2+
einops==0.3.0
3+
huggingface-hub==0.8.1
4+
imageio-ffmpeg==0.4.2
5+
imageio==2.9.0
6+
kornia==0.6.0
7+
# pip will resolve the version which matches torch
8+
numpy
9+
omegaconf==2.1.1
10+
opencv-python==4.6.0.66
11+
pillow==9.2.0
12+
pip>=22
13+
pudb==2019.2
14+
pytorch-lightning==1.4.2
15+
streamlit==1.12.0
16+
# "CompVis/taming-transformers" doesn't work
17+
# ldm\models\autoencoder.py", line 6, in <module>
18+
# from taming.modules.vqvae.quantize import VectorQuantizer2 as VectorQuantizer
19+
# ModuleNotFoundError
20+
taming-transformers-rom1504==0.0.6
21+
test-tube>=0.7.5
22+
torch-fidelity==0.3.0
23+
torchmetrics==0.6.0
24+
transformers==4.19.2
25+
git+https://github.com/openai/CLIP.git@main#egg=clip
26+
git+https://github.com/lstein/k-diffusion.git@master#egg=k-diffusion
27+
# No CUDA in PyPi builds
28+
--extra-index-url https://download.pytorch.org/whl/cu113 --trusted-host https://download.pytorch.org
29+
torch==1.11.0
30+
# Same as numpy - let pip do its thing
31+
torchvision
32+
-e .
File renamed without changes.

requirements-win.txt

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
albumentations==0.4.3
2+
einops==0.3.0
3+
huggingface-hub==0.8.1
4+
imageio-ffmpeg==0.4.2
5+
imageio==2.9.0
6+
kornia==0.6.0
7+
# pip will resolve the version which matches torch
8+
numpy
9+
omegaconf==2.1.1
10+
opencv-python==4.6.0.66
11+
pillow==9.2.0
12+
pip>=22
13+
pudb==2019.2
14+
pytorch-lightning==1.4.2
15+
streamlit==1.12.0
16+
# "CompVis/taming-transformers" doesn't work
17+
# ldm\models\autoencoder.py", line 6, in <module>
18+
# from taming.modules.vqvae.quantize import VectorQuantizer2 as VectorQuantizer
19+
# ModuleNotFoundError
20+
taming-transformers-rom1504==0.0.6
21+
test-tube>=0.7.5
22+
torch-fidelity==0.3.0
23+
torchmetrics==0.6.0
24+
transformers==4.19.2
25+
git+https://github.com/openai/CLIP.git@main#egg=clip
26+
git+https://github.com/lstein/k-diffusion.git@master#egg=k-diffusion
27+
# No CUDA in PyPi builds
28+
--extra-index-url https://download.pytorch.org/whl/cu113 --trusted-host https://download.pytorch.org
29+
torch==1.11.0
30+
# Same as numpy - let pip do its thing
31+
torchvision
32+
-e .

0 commit comments

Comments
 (0)