mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2024-12-27 03:44:35 +00:00
f4183afe6a
* scripts : add --outdir option to hf.sh This commit adds an option to the hf.sh script that allows the user to specify an output directory for the downloaded file. The motivation for this changes is that examples that use the hf.sh script to download models from huggingface can now specify the output directory, perhaps to the `models` directory to keep them in one place and not clutter the root directory. Signed-off-by: Daniel Bevenius <daniel.bevenius@gmail.com> * squash! scripts : add --outdir option to hf.sh Fix format of the --outdir option in the usage message. Signed-off-by: Daniel Bevenius <daniel.bevenius@gmail.com> --------- Signed-off-by: Daniel Bevenius <daniel.bevenius@gmail.com>
113 lines
2.3 KiB
Bash
Executable File
113 lines
2.3 KiB
Bash
Executable File
#!/bin/bash
|
|
#
|
|
# Shortcut for downloading HF models
|
|
#
|
|
# Usage:
|
|
# ./main -m $(./scripts/hf.sh https://huggingface.co/TheBloke/Mixtral-8x7B-v0.1-GGUF/resolve/main/mixtral-8x7b-v0.1.Q4_K_M.gguf)
|
|
# ./main -m $(./scripts/hf.sh --url https://huggingface.co/TheBloke/Mixtral-8x7B-v0.1-GGUF/blob/main/mixtral-8x7b-v0.1.Q4_K_M.gguf)
|
|
# ./main -m $(./scripts/hf.sh --repo TheBloke/Mixtral-8x7B-v0.1-GGUF --file mixtral-8x7b-v0.1.Q4_K_M.gguf)
|
|
#
|
|
|
|
# all logs go to stderr
|
|
function log {
|
|
echo "$@" 1>&2
|
|
}
|
|
|
|
function usage {
|
|
log "Usage: $0 [[--url] <url>] [--repo <repo>] [--file <file>] [--outdir <dir> [-h|--help]"
|
|
exit 1
|
|
}
|
|
|
|
# check for curl or wget
|
|
function has_cmd {
|
|
if ! [ -x "$(command -v $1)" ]; then
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
if has_cmd wget; then
|
|
cmd="wget -q --show-progress -c -O %s/%s %s"
|
|
elif has_cmd curl; then
|
|
cmd="curl -C - -f --output-dir %s -o %s -L %s"
|
|
else
|
|
log "[E] curl or wget not found"
|
|
exit 1
|
|
fi
|
|
|
|
url=""
|
|
repo=""
|
|
file=""
|
|
outdir="."
|
|
|
|
# parse args
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
--url)
|
|
url="$2"
|
|
shift 2
|
|
;;
|
|
--repo)
|
|
repo="$2"
|
|
shift 2
|
|
;;
|
|
--file)
|
|
file="$2"
|
|
shift 2
|
|
;;
|
|
--outdir)
|
|
outdir="$2"
|
|
shift 2
|
|
;;
|
|
-h|--help)
|
|
usage
|
|
;;
|
|
*)
|
|
url="$1"
|
|
shift
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if [ -n "$repo" ] && [ -n "$file" ]; then
|
|
url="https://huggingface.co/$repo/resolve/main/$file"
|
|
fi
|
|
|
|
if [ -z "$url" ]; then
|
|
log "[E] missing --url"
|
|
usage
|
|
fi
|
|
|
|
# check if the URL is a HuggingFace model, and if so, try to download it
|
|
is_url=false
|
|
|
|
if [[ ${#url} -gt 22 ]]; then
|
|
if [[ ${url:0:22} == "https://huggingface.co" ]]; then
|
|
is_url=true
|
|
fi
|
|
fi
|
|
|
|
if [ "$is_url" = false ]; then
|
|
log "[E] invalid URL, must start with https://huggingface.co"
|
|
exit 0
|
|
fi
|
|
|
|
# replace "blob/main" with "resolve/main"
|
|
url=${url/blob\/main/resolve\/main}
|
|
|
|
basename=$(basename $url)
|
|
|
|
log "[+] attempting to download $basename"
|
|
|
|
if [ -n "$cmd" ]; then
|
|
cmd=$(printf "$cmd" "$outdir" "$basename" "$url")
|
|
log "[+] $cmd"
|
|
if $cmd; then
|
|
echo $outdir/$basename
|
|
exit 0
|
|
fi
|
|
fi
|
|
|
|
log "[-] failed to download"
|
|
|
|
exit 1
|