|
|
@@ -3,9 +3,9 @@
|
|
|
# Shortcut for downloading HF models
|
|
|
#
|
|
|
# Usage:
|
|
|
-# ./main -m $(./examples/hf.sh https://huggingface.co/TheBloke/Mixtral-8x7B-v0.1-GGUF/resolve/main/mixtral-8x7b-v0.1.Q4_K_M.gguf)
|
|
|
-# ./main -m $(./examples/hf.sh --url https://huggingface.co/TheBloke/Mixtral-8x7B-v0.1-GGUF/blob/main/mixtral-8x7b-v0.1.Q4_K_M.gguf)
|
|
|
-# ./main -m $(./examples/hf.sh --repo TheBloke/Mixtral-8x7B-v0.1-GGUF --file mixtral-8x7b-v0.1.Q4_K_M.gguf)
|
|
|
+# ./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
|
|
|
@@ -14,7 +14,7 @@ function log {
|
|
|
}
|
|
|
|
|
|
function usage {
|
|
|
- log "Usage: $0 [[--url] <url>] [--repo <repo>] [--file <file>] [-h|--help]"
|
|
|
+ log "Usage: $0 [[--url] <url>] [--repo <repo>] [--file <file>] [--outdir <dir> [-h|--help]"
|
|
|
exit 1
|
|
|
}
|
|
|
|
|
|
@@ -26,9 +26,9 @@ function has_cmd {
|
|
|
}
|
|
|
|
|
|
if has_cmd wget; then
|
|
|
- cmd="wget -q --show-progress -c -O %s %s"
|
|
|
+ cmd="wget -q --show-progress -c -O %s/%s %s"
|
|
|
elif has_cmd curl; then
|
|
|
- cmd="curl -C - -f -o %s -L %s"
|
|
|
+ cmd="curl -C - -f --output-dir %s -o %s -L %s"
|
|
|
else
|
|
|
log "[E] curl or wget not found"
|
|
|
exit 1
|
|
|
@@ -37,6 +37,7 @@ fi
|
|
|
url=""
|
|
|
repo=""
|
|
|
file=""
|
|
|
+outdir="."
|
|
|
|
|
|
# parse args
|
|
|
while [[ $# -gt 0 ]]; do
|
|
|
@@ -53,6 +54,10 @@ while [[ $# -gt 0 ]]; do
|
|
|
file="$2"
|
|
|
shift 2
|
|
|
;;
|
|
|
+ --outdir)
|
|
|
+ outdir="$2"
|
|
|
+ shift 2
|
|
|
+ ;;
|
|
|
-h|--help)
|
|
|
usage
|
|
|
;;
|
|
|
@@ -94,10 +99,10 @@ basename=$(basename $url)
|
|
|
log "[+] attempting to download $basename"
|
|
|
|
|
|
if [ -n "$cmd" ]; then
|
|
|
- cmd=$(printf "$cmd" "$basename" "$url")
|
|
|
+ cmd=$(printf "$cmd" "$outdir" "$basename" "$url")
|
|
|
log "[+] $cmd"
|
|
|
if $cmd; then
|
|
|
- echo $basename
|
|
|
+ echo $outdir/$basename
|
|
|
exit 0
|
|
|
fi
|
|
|
fi
|