AlexKM commited on
Commit
09b516c
1 Parent(s): 711d859

Upload download_models.sh

Browse files
Files changed (1) hide show
  1. download_models.sh +111 -0
download_models.sh ADDED
@@ -0,0 +1,111 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+ LOC=checkpoints
3
+ mkdir -p "$LOC"
4
+
5
+ # Which models to download?
6
+ IMAGENET_1024=false
7
+ IMAGENET_16384=true
8
+ GUMBEL=false
9
+ #WIKIART_1024=false
10
+ WIKIART_16384=false
11
+ COCO=false
12
+ FACESHQ=false
13
+ SFLCKR=false
14
+
15
+
16
+ if [ "$IMAGENET_1024" = true ] ; then
17
+ # imagenet_1024 - 958 MB:
18
+ if [ ! -f "$LOC"/vqgan_imagenet_f16_1024.yaml ]; then
19
+ curl -L -o "$LOC"/vqgan_imagenet_f16_1024.yaml -C - 'https://heibox.uni-heidelberg.de/d/8088892a516d4e3baf92/files/?p=%2Fconfigs%2Fmodel.yaml&dl=1' #ImageNet 1024
20
+ fi
21
+ if [ ! -f "$LOC"/vqgan_imagenet_f16_1024.ckpt ]; then
22
+ curl -L -o "$LOC"/vqgan_imagenet_f16_1024.ckpt -C - 'https://heibox.uni-heidelberg.de/d/8088892a516d4e3baf92/files/?p=%2Fckpts%2Flast.ckpt&dl=1' #ImageNet 1024
23
+ fi
24
+ fi
25
+
26
+ if [ "$IMAGENET_16384" = true ] ; then
27
+ # imagenet_16384 - 980 MB:
28
+ if [ ! -f "$LOC"/vqgan_imagenet_f16_16384.yaml ]; then
29
+ curl -L -o "$LOC"/vqgan_imagenet_f16_16384.yaml -C - 'https://heibox.uni-heidelberg.de/d/a7530b09fed84f80a887/files/?p=%2Fconfigs%2Fmodel.yaml&dl=1' #ImageNet 16384
30
+ fi
31
+ if [ ! -f "$LOC"/vqgan_imagenet_f16_16384.ckpt ]; then
32
+ curl -L -o "$LOC"/vqgan_imagenet_f16_16384.ckpt -C - 'https://heibox.uni-heidelberg.de/d/a7530b09fed84f80a887/files/?p=%2Fckpts%2Flast.ckpt&dl=1' #ImageNet 16384
33
+ fi
34
+ fi
35
+
36
+ if [ "$GUMBEL" = true ] ; then
37
+ # vqgan_gumbel_f8_8192 (was openimages_f16_8192) - 376 MB:
38
+ if [ ! -f "$LOC"/vqgan_gumbel_f8_8192.yaml ]; then
39
+ curl -L -o "$LOC"/vqgan_gumbel_f8_8192.yaml -C - 'https://heibox.uni-heidelberg.de/d/2e5662443a6b4307b470/files/?p=%2Fconfigs%2Fmodel.yaml&dl=1'
40
+ fi
41
+ if [ ! -f "$LOC"/vqgan_gumbel_f8_8192.ckpt ]; then
42
+ curl -L -o "$LOC"/vqgan_gumbel_f8_8192.ckpt -C - 'https://heibox.uni-heidelberg.de/d/2e5662443a6b4307b470/files/?p=%2Fckpts%2Flast.ckpt&dl=1'
43
+ fi
44
+ fi
45
+
46
+ if [ "$COCO" = true ] ; then
47
+ # coco - 8.4 GB:
48
+ if [ ! -f "$LOC"/coco.yaml ]; then
49
+ curl -L -o "$LOC"/coco.yaml -C - 'https://dl.nmkd.de/ai/clip/coco/coco.yaml' #COCO
50
+ fi
51
+ if [ ! -f "$LOC"/coco.ckpt ]; then
52
+ curl -L -o "$LOC"/coco.ckpt -C - 'https://dl.nmkd.de/ai/clip/coco/coco.ckpt' #COCO
53
+ fi
54
+ fi
55
+
56
+ if [ "$FACESHQ" = true ] ; then
57
+ # faceshq:
58
+ if [ ! -f "$LOC"/faceshq.yaml ]; then
59
+ curl -L -o "$LOC"/faceshq.yaml -C - 'https://drive.google.com/uc?export=download&id=1fHwGx_hnBtC8nsq7hesJvs-Klv-P0gzT' #FacesHQ
60
+ fi
61
+ if [ ! -f "$LOC"/faceshq.ckpt ]; then
62
+ curl -L -o "$LOC"/faceshq.ckpt -C - 'https://app.koofr.net/content/links/a04deec9-0c59-4673-8b37-3d696fe63a5d/files/get/last.ckpt?path=%2F2020-11-13T21-41-45_faceshq_transformer%2Fcheckpoints%2Flast.ckpt' #FacesHQ
63
+ fi
64
+ fi
65
+
66
+ # Link?
67
+ #if [ "$WIKIART_1024" = true ] ; then
68
+ # # wikiart_1024 - 958 MB:
69
+ # if [ ! -f "$LOC"/wikiart_1024.yaml ]; then
70
+ # curl -L -o "$LOC"/wikiart_1024.yaml -C - 'http://mirror.io.community/blob/vqgan/wikiart.yaml' #WikiArt 1024
71
+ # fi
72
+ # if [ ! -f "$LOC"/wikiart_1024.ckpt ]; then
73
+ # curl -L -o "$LOC"/wikiart_1024.ckpt -C - 'http://mirror.io.community/blob/vqgan/wikiart.ckpt' #WikiArt 1024
74
+ # fi
75
+ #fi
76
+
77
+ if [ "$WIKIART_16384" = true ] ; then
78
+ #wikiart_16384 - 1 GB:
79
+ if [ ! -f "$LOC"/wikiart_16384.yaml ]; then
80
+ curl -L -o "$LOC"/wikiart_16384.yaml -C - 'http://eaidata.bmk.sh/data/Wikiart_16384/wikiart_f16_16384_8145600.yaml' #WikiArt 16384
81
+ fi
82
+ if [ ! -f "$LOC"/wikiart_16384.ckpt ]; then
83
+ curl -L -o "$LOC"/wikiart_16384.ckpt -C - 'http://eaidata.bmk.sh/data/Wikiart_16384/wikiart_f16_16384_8145600.ckpt' #WikiArt 16384
84
+ fi
85
+ fi
86
+
87
+ if [ "$SFLCKR" = true ] ; then
88
+ # sflckr:
89
+ if [ ! -f "$LOC"/sflckr.yaml ]; then
90
+ curl -L -o "$LOC"/sflckr.yaml -C - 'https://heibox.uni-heidelberg.de/d/73487ab6e5314cb5adba/files/?p=%2Fconfigs%2F2020-11-09T13-31-51-project.yaml&dl=1' #S-FLCKR
91
+ fi
92
+ if [ ! -f "$LOC"/sflckr.ckpt ]; then
93
+ curl -L -o "$LOC"/sflckr.ckpt -C - 'https://heibox.uni-heidelberg.de/d/73487ab6e5314cb5adba/files/?p=%2Fcheckpoints%2Flast.ckpt&dl=1' #S-FLCKR
94
+ fi
95
+ fi
96
+
97
+ # Others:
98
+
99
+ #ade20k:
100
+ # curl -L -o ade20k.yaml -C - 'https://static.miraheze.org/intercriaturaswiki/b/bf/Ade20k.txt' #ADE20K
101
+ # curl -L -o ade20k.ckpt -C - 'https://app.koofr.net/content/links/0f65c2cd-7102-4550-a2bd-07fd383aac9e/files/get/last.ckpt?path=%2F2020-11-20T21-45-44_ade20k_transformer%2Fcheckpoints%2Flast.ckpt' #ADE20K
102
+
103
+ #ffhq:
104
+ # curl -L -o ffhq.yaml -C - 'https://app.koofr.net/content/links/0fc005bf-3dca-4079-9d40-cdf38d42cd7a/files/get/2021-04-23T18-19-01-project.yaml?path=%2F2021-04-23T18-19-01_ffhq_transformer%2Fconfigs%2F2021-04-23T18-19-01-project.yaml&force' #FFHQ
105
+ # curl -L -o ffhq.ckpt -C - 'https://app.koofr.net/content/links/0fc005bf-3dca-4079-9d40-cdf38d42cd7a/files/get/last.ckpt?path=%2F2021-04-23T18-19-01_ffhq_transformer%2Fcheckpoints%2Flast.ckpt&force' #FFHQ
106
+
107
+ #celebahq:
108
+ # curl -L -o celebahq.yaml -C - 'https://app.koofr.net/content/links/6dddf083-40c8-470a-9360-a9dab2a94e96/files/get/2021-04-23T18-11-19-project.yaml?path=%2F2021-04-23T18-11-19_celebahq_transformer%2Fconfigs%2F2021-04-23T18-11-19-project.yaml&force' #CelebA-HQ
109
+ # curl -L -o celebahq.ckpt -C - 'https://app.koofr.net/content/links/6dddf083-40c8-470a-9360-a9dab2a94e96/files/get/last.ckpt?path=%2F2021-04-23T18-11-19_celebahq_transformer%2Fcheckpoints%2Flast.ckpt&force' #CelebA-HQ
110
+
111
+ #