Simple script to download a model from civitai.com - using bash and jq

civit.sh

#!/bin/bash
# save this file in ~/.local/bin/civit, make it executable
# usage: civit <URN> [path}
#
# Author: Patrice Ferlet <[email protected]>
# License: BSDv3
#
# Please ensure you've installed "jq" command and "curl"
# The URN can be easily copied from the model page, on the right, the "AIR"
# This script will download in the provided direcoty (default to the current directory)
# and checks the SHA256 sum to prevent corrupted files.
#
# See the bottom of the script for an example.

URN=${1}
OUTPUT_DIR=${2-./}

# assert that URN is provided
[ -z "$URN" ] && echo "URN is empty, please give it as first argument" && exit 1

# extract MODELID and VERSIONID
MODELID=${URN##*:}
MODELID=${MODELID%%@*}
VERSIONID=${URN##*@}

# assert we've got $MODELID and $VERSIONID
[ -z "$MODELID" ] && echo "MODELID is empty" && exit 1
[ -z "$VERSIONID" ] && echo "VERSIONID is empty" && exit 1

# Token
[ -f ~/.local/share/civitai.token ] && API_KEY=$(cat ~/.local/share/civitai.token)
[ -f ~/.config/civitai.token ] && API_KEY=$(cat ~/.config/civitai.token)
[ -z "$API_KEY" ] && echo "Please, save your civitai token in ~/.local/share/civitai.token or ~/.config/civitai.token" && exit 1

# reading JSON
APIURL=https://civitai.com/api/v1/models/${MODELID}
CMD=$(curl -fL -sS ${APIURL} | jq -r \
  '.modelVersions[] | select (.id == '${VERSIONID}') | .files[0] |
    ("NAME=\"" + .name + "\";URL=\"" + .downloadUrl + "\";SHA256=\"" + .hashes.SHA256 + "\"")')

# eval $CMD to set NAME, URL, SHA256
eval $CMD

# Ensure we have NAME, URL, SHA256
[ -z "$NAME" ] && echo "NAME is empty" && exit 1
[ -z "$URL" ] && echo "URL is empty" && exit 1
[ -z "$SHA256" ] && echo "SHA256 is empty" && exit 1

# READ the civitai
if [ ! -f "${OUTPUT_DIR}/${NAME}" ]; then
  echo "📥 Downloading ${NAME} to ${OUTPUT_DIR} ..."
  curl -Lf --progress-bar -H "Authorization: Bearer $API_KEY" "${URL}" >"$OUTPUT_DIR/${NAME}"
  echo "Done"
else
  echo "🥳 File already exists, skipping download"
fi

# esnure SHA256
echo "👁️ Verifying SHA256..."
echo $SHA256 "$OUTPUT_DIR/${NAME}" | sha256sum --check --status && echo "✅ Good, the model is ready to use" || echo "❌ SHA256 doesn't match, the file is potentially corrupted"

# exemple:
# $ civit urn:air:sdxl:checkpoint:civitai:112902@354657
# 📥 Downloading dreamshaperXL_lightningDPMSDE.safetensors to ./ ...
# ##################################################################################################################################################################### 100.0%
# Done
# 👁️ Verifying SHA256...
# ✅ Good, the model is ready to use