Upload folder using huggingface_hub

Stable-Cascade-SR/models/download_models.sh

@@ -0,0 +1,70 @@
+#!/bin/bash
+
+# Check if at least two arguments were provided (excluding the optional first one)
+if [ $# -lt 2 ]; then
+    echo "Insufficient arguments provided. At least two arguments are required."
+    exit 1
+fi
+
+# Check for the optional "essential" argument and download the essential models if present
+if [ "$1" == "essential" ]; then
+        echo "Downloading Essential Models (EfficientNet, Stage A, Previewer)"
+        wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_a.safetensors -P . -q --show-progress
+        wget https://huggingface.co/stabilityai/StableWurst/resolve/main/previewer.safetensors -P . -q --show-progress
+        wget https://huggingface.co/stabilityai/StableWurst/resolve/main/effnet_encoder.safetensors -P . -q --show-progress
+    shift # Move the arguments, $2 becomes $1, $3 becomes $2, etc.
+fi
+
+# Now, $1 is the second argument due to the potential shift above
+second_argument="$1"
+binary_decision="${2:-bfloat16}" # Use default or specific binary value if provided
+
+case $second_argument in
+    big-big)
+        if [ "$binary_decision" == "bfloat16" ]; then
+            echo "Downloading Large Stage B & Large Stage C"
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b_bf16.safetensors -P . -q --show-progress
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c_bf16.safetensors -P . -q --show-progress
+        else
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b.safetensors -P . -q --show-progress
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c.safetensors -P . -q --show-progress
+        fi
+        ;;
+    big-small)
+        if [ "$binary_decision" == "bfloat16" ]; then
+            echo "Downloading Large Stage B & Small Stage C (BFloat16)"
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b_bf16.safetensors -P . -q --show-progress
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c_lite_bf16.safetensors -P . -q --show-progress
+        else
+            echo "Downloading Large Stage B & Small Stage C"
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b.safetensors -P . -q --show-progress
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c_lite.safetensors -P . -q --show-progress
+        fi
+        ;;
+    small-big)
+        if [ "$binary_decision" == "bfloat16" ]; then
+            echo "Downloading Small Stage B & Large Stage C (BFloat16)"
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b_lite_bf16.safetensors -P . -q --show-progress
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c_bf16.safetensors -P . -q --show-progress
+        else
+            echo "Downloading Small Stage B & Large Stage C"
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b_lite.safetensors -P . -q --show-progress
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c.safetensors -P . -q --show-progress
+        fi
+        ;;
+    small-small)
+        if [ "$binary_decision" == "bfloat16" ]; then
+            echo "Downloading Small Stage B & Small Stage C (BFloat16)"
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b_lite_bf16.safetensors -P . -q --show-progress
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c_lite_bf16.safetensors -P . -q --show-progress
+        else
+            echo "Downloading Small Stage B & Small Stage C"
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b_lite.safetensors -P . -q --show-progress
+            wget https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c_lite.safetensors -P . -q --show-progress
+        fi
+        ;;
+    *)
+        echo "Invalid second argument. Please provide a valid argument: big-big, big-small, small-big, or small-small."
+        exit 2
+        ;;
+esac

Stable-Cascade-SR/models/effnet_encoder.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e23c1fe17ba8cdd6c49a5b292d72502bd4cf729e7de9f2060bb4dcc26737c0a0
+size 81506763

Stable-Cascade-SR/models/inpainting.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e115e477addc7ad34156a743a922ef8bef248be791d3846008a2220a198da7c4
+size 217741855

Stable-Cascade-SR/models/previewer.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7d66285b840aaaf69abe809727f0b02657c513dd9376f2f07f3c3ba4fe87f250
+size 15954183

Stable-Cascade-SR/models/readme.md

@@ -0,0 +1,40 @@
+# Download Models
+
+As there are many models provided, let's make sure you only download the ones you need.
+The ``download_models.sh`` will make that very easy. The basic usage looks like this: <br>
+```bash
+bash download_models.sh essential variant bfloat16
+```
+
+**essential**<br>
+This is optional and determines if you want to download the EfficientNet, Stage A & Previewer. 
+If this is the first time you run this command, you should definitely do it, because we need it.
+
+**variant**<br>
+This determines which varient you want to use for **Stage B** and **Stage C**.
+There are four options:
+
+|                     | Stage C (Large) | Stage C (Lite) |
+|---------------------|-----------------|----------------|
+| **Stage B (Large)** | big-big         | big-small      |
+| **Stage B (Lite)**  | small-big       | small-small    |
+
+
+So if you want to download the large Stage B & large Stage C you can execute: <br> 
+```bash
+bash download_models.sh essential big-big bfloat16
+```
+
+**bfloat16** <br>
+The last argument is optional as well, and simply determines in which precision you download Stage B & Stage C.
+If you want a faster download, choose _bfloat16_ (if your machine supports it), otherwise use _float32_.
+
+### Recommendation
+If your GPU allows for it, you should definitely go for the **large** Stage C, which has 3.6 billion parameters.
+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.
+For Stage B the difference is not so big. The **large** Stage B is better at reconstructing small details,
+but if your GPU is not so powerful, just go for the smaller one.
+
+### Remark
+Unfortunately, you can not run the models in float16 at the moment. Only bfloat16 or float32 work for now. However,
+with some investigation, it should be possible to fix the overflowing and allow for inference in float16 as well.

Stable-Cascade-SR/models/stage_a.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0a3f5a0b618494f81964dc719dbaaa9bd52f207d464d5fbf45442d363e1c0b5a
+size 73654345

Stable-Cascade-SR/models/stage_b_bf16.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2c8d58b267678aecfa6705a0a0375c88613065a8a8d32ad3a4c3867f5461cb3a
+size 3126069112

Stable-Cascade-SR/models/stage_c_bf16.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dfe24009fc881011f350d08d9d13be13a1a3b3cbfed667435efe0fd419aca099
+size 7178375922

Stable-Cascade-SR/models/super_resolution.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:91aa02e90bce912fcb00dad4c864be9c3f31737ce4239f8276418f48954c2f42
+size 813989376