File size: 4,095 Bytes
1541adc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
#!/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
|