ameerazam08
commited on
Commit
•
af10e49
1
Parent(s):
065e593
Upload folder using huggingface_hub
Browse files- models/download_models.sh +70 -0
- models/effnet_encoder.safetensors +3 -0
- models/inpainting.safetensors +3 -0
- models/previewer.safetensors +3 -0
- models/readme.md +40 -0
- models/stage_a.safetensors +3 -0
- models/stage_b_bf16.safetensors +3 -0
- models/stage_c_bf16.safetensors +3 -0
- models/super_resolution.safetensors +3 -0
models/download_models.sh
ADDED
@@ -0,0 +1,70 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
#!/bin/bash
|
2 |
+
|
3 |
+
# Check if at least two arguments were provided (excluding the optional first one)
|
4 |
+
if [ $# -lt 2 ]; then
|
5 |
+
echo "Insufficient arguments provided. At least two arguments are required."
|
6 |
+
exit 1
|
7 |
+
fi
|
8 |
+
|
9 |
+
# Check for the optional "essential" argument and download the essential models if present
|
10 |
+
if [ "$1" == "essential" ]; then
|
11 |
+
echo "Downloading Essential Models (EfficientNet, Stage A, Previewer)"
|
12 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_a.safetensors -P . -q --show-progress
|
13 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/previewer.safetensors -P . -q --show-progress
|
14 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/effnet_encoder.safetensors -P . -q --show-progress
|
15 |
+
shift # Move the arguments, $2 becomes $1, $3 becomes $2, etc.
|
16 |
+
fi
|
17 |
+
|
18 |
+
# Now, $1 is the second argument due to the potential shift above
|
19 |
+
second_argument="$1"
|
20 |
+
binary_decision="${2:-bfloat16}" # Use default or specific binary value if provided
|
21 |
+
|
22 |
+
case $second_argument in
|
23 |
+
big-big)
|
24 |
+
if [ "$binary_decision" == "bfloat16" ]; then
|
25 |
+
echo "Downloading Large Stage B & Large Stage C"
|
26 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b_bf16.safetensors -P . -q --show-progress
|
27 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c_bf16.safetensors -P . -q --show-progress
|
28 |
+
else
|
29 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b.safetensors -P . -q --show-progress
|
30 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c.safetensors -P . -q --show-progress
|
31 |
+
fi
|
32 |
+
;;
|
33 |
+
big-small)
|
34 |
+
if [ "$binary_decision" == "bfloat16" ]; then
|
35 |
+
echo "Downloading Large Stage B & Small Stage C (BFloat16)"
|
36 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b_bf16.safetensors -P . -q --show-progress
|
37 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c_lite_bf16.safetensors -P . -q --show-progress
|
38 |
+
else
|
39 |
+
echo "Downloading Large Stage B & Small Stage C"
|
40 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b.safetensors -P . -q --show-progress
|
41 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c_lite.safetensors -P . -q --show-progress
|
42 |
+
fi
|
43 |
+
;;
|
44 |
+
small-big)
|
45 |
+
if [ "$binary_decision" == "bfloat16" ]; then
|
46 |
+
echo "Downloading Small Stage B & Large Stage C (BFloat16)"
|
47 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b_lite_bf16.safetensors -P . -q --show-progress
|
48 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c_bf16.safetensors -P . -q --show-progress
|
49 |
+
else
|
50 |
+
echo "Downloading Small Stage B & Large Stage C"
|
51 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b_lite.safetensors -P . -q --show-progress
|
52 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c.safetensors -P . -q --show-progress
|
53 |
+
fi
|
54 |
+
;;
|
55 |
+
small-small)
|
56 |
+
if [ "$binary_decision" == "bfloat16" ]; then
|
57 |
+
echo "Downloading Small Stage B & Small Stage C (BFloat16)"
|
58 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b_lite_bf16.safetensors -P . -q --show-progress
|
59 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c_lite_bf16.safetensors -P . -q --show-progress
|
60 |
+
else
|
61 |
+
echo "Downloading Small Stage B & Small Stage C"
|
62 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b_lite.safetensors -P . -q --show-progress
|
63 |
+
wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c_lite.safetensors -P . -q --show-progress
|
64 |
+
fi
|
65 |
+
;;
|
66 |
+
*)
|
67 |
+
echo "Invalid second argument. Please provide a valid argument: big-big, big-small, small-big, or small-small."
|
68 |
+
exit 2
|
69 |
+
;;
|
70 |
+
esac
|
models/effnet_encoder.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e23c1fe17ba8cdd6c49a5b292d72502bd4cf729e7de9f2060bb4dcc26737c0a0
|
3 |
+
size 81506763
|
models/inpainting.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e115e477addc7ad34156a743a922ef8bef248be791d3846008a2220a198da7c4
|
3 |
+
size 217741855
|
models/previewer.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7d66285b840aaaf69abe809727f0b02657c513dd9376f2f07f3c3ba4fe87f250
|
3 |
+
size 15954183
|
models/readme.md
ADDED
@@ -0,0 +1,40 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Download Models
|
2 |
+
|
3 |
+
As there are many models provided, let's make sure you only download the ones you need.
|
4 |
+
The ``download_models.sh`` will make that very easy. The basic usage looks like this: <br>
|
5 |
+
```bash
|
6 |
+
bash download_models.sh essential variant bfloat16
|
7 |
+
```
|
8 |
+
|
9 |
+
**essential**<br>
|
10 |
+
This is optional and determines if you want to download the EfficientNet, Stage A & Previewer.
|
11 |
+
If this is the first time you run this command, you should definitely do it, because we need it.
|
12 |
+
|
13 |
+
**variant**<br>
|
14 |
+
This determines which varient you want to use for **Stage B** and **Stage C**.
|
15 |
+
There are four options:
|
16 |
+
|
17 |
+
| | Stage C (Large) | Stage C (Lite) |
|
18 |
+
|---------------------|-----------------|----------------|
|
19 |
+
| **Stage B (Large)** | big-big | big-small |
|
20 |
+
| **Stage B (Lite)** | small-big | small-small |
|
21 |
+
|
22 |
+
|
23 |
+
So if you want to download the large Stage B & large Stage C you can execute: <br>
|
24 |
+
```bash
|
25 |
+
bash download_models.sh essential big-big bfloat16
|
26 |
+
```
|
27 |
+
|
28 |
+
**bfloat16** <br>
|
29 |
+
The last argument is optional as well, and simply determines in which precision you download Stage B & Stage C.
|
30 |
+
If you want a faster download, choose _bfloat16_ (if your machine supports it), otherwise use _float32_.
|
31 |
+
|
32 |
+
### Recommendation
|
33 |
+
If your GPU allows for it, you should definitely go for the **large** Stage C, which has 3.6 billion parameters.
|
34 |
+
It is a lot better and was finetuned a lot more. Also, the ControlNet and Lora examples are only for the large Stage C at the moment.
|
35 |
+
For Stage B the difference is not so big. The **large** Stage B is better at reconstructing small details,
|
36 |
+
but if your GPU is not so powerful, just go for the smaller one.
|
37 |
+
|
38 |
+
### Remark
|
39 |
+
Unfortunately, you can not run the models in float16 at the moment. Only bfloat16 or float32 work for now. However,
|
40 |
+
with some investigation, it should be possible to fix the overflowing and allow for inference in float16 as well.
|
models/stage_a.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0a3f5a0b618494f81964dc719dbaaa9bd52f207d464d5fbf45442d363e1c0b5a
|
3 |
+
size 73654345
|
models/stage_b_bf16.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2c8d58b267678aecfa6705a0a0375c88613065a8a8d32ad3a4c3867f5461cb3a
|
3 |
+
size 3126069112
|
models/stage_c_bf16.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dfe24009fc881011f350d08d9d13be13a1a3b3cbfed667435efe0fd419aca099
|
3 |
+
size 7178375922
|
models/super_resolution.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:91aa02e90bce912fcb00dad4c864be9c3f31737ce4239f8276418f48954c2f42
|
3 |
+
size 813989376
|