scripts : fix [no ci]
This commit is contained in:
parent
414a66f6b1
commit
3e794f2eba
2 changed files with 3 additions and 115 deletions
112
scripts/hf.sh
112
scripts/hf.sh
|
@ -1,112 +0,0 @@
|
||||||
#!/bin/bash
|
|
||||||
#
|
|
||||||
# Shortcut for downloading HF models
|
|
||||||
#
|
|
||||||
# Usage:
|
|
||||||
# ./llama-cli -m $(./scripts/hf.sh https://huggingface.co/TheBloke/Mixtral-8x7B-v0.1-GGUF/resolve/main/mixtral-8x7b-v0.1.Q4_K_M.gguf)
|
|
||||||
# ./llama-cli -m $(./scripts/hf.sh --url https://huggingface.co/TheBloke/Mixtral-8x7B-v0.1-GGUF/blob/main/mixtral-8x7b-v0.1.Q4_K_M.gguf)
|
|
||||||
# ./llama-cli -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 -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
|
|
|
@ -80,18 +80,18 @@ run_conversion_and_inference_lora() {
|
||||||
# Run inference
|
# Run inference
|
||||||
echo -e "\n\n---------------------------\n\n"
|
echo -e "\n\n---------------------------\n\n"
|
||||||
echo "Running llama-cli without lora for $model_name with hidden_size $hidden_size..."
|
echo "Running llama-cli without lora for $model_name with hidden_size $hidden_size..."
|
||||||
OUTPUT_BASE=$(./llama-cli -m $MODELS_REPO/$model_name/hidden_size=$hidden_size/base/Base-F32.gguf \
|
OUTPUT_BASE=$(./llama-cli -no-cnv -m $MODELS_REPO/$model_name/hidden_size=$hidden_size/base/Base-F32.gguf \
|
||||||
-p "$EXPECTED_BASE_FIRST_WORD" -n 50 --seed 42 --temp 0)
|
-p "$EXPECTED_BASE_FIRST_WORD" -n 50 --seed 42 --temp 0)
|
||||||
|
|
||||||
echo -e "\n\n---------------------------\n\n"
|
echo -e "\n\n---------------------------\n\n"
|
||||||
echo "Running llama-cli with hot lora for $model_name with hidden_size $hidden_size..."
|
echo "Running llama-cli with hot lora for $model_name with hidden_size $hidden_size..."
|
||||||
OUTPUT_LORA_HOT=$(./llama-cli -m $MODELS_REPO/$model_name/hidden_size=$hidden_size/base/Base-F32.gguf \
|
OUTPUT_LORA_HOT=$(./llama-cli -no-cnv -m $MODELS_REPO/$model_name/hidden_size=$hidden_size/base/Base-F32.gguf \
|
||||||
--lora $MODELS_REPO/$model_name/hidden_size=$hidden_size/lora/Lora-F32-LoRA.gguf \
|
--lora $MODELS_REPO/$model_name/hidden_size=$hidden_size/lora/Lora-F32-LoRA.gguf \
|
||||||
-p "$EXPECTED_LORA_FIRST_WORD" -n 50 --seed 42 --temp 0)
|
-p "$EXPECTED_LORA_FIRST_WORD" -n 50 --seed 42 --temp 0)
|
||||||
|
|
||||||
echo -e "\n\n---------------------------\n\n"
|
echo -e "\n\n---------------------------\n\n"
|
||||||
echo "Running llama-cli with merged lora for $model_name with hidden_size $hidden_size..."
|
echo "Running llama-cli with merged lora for $model_name with hidden_size $hidden_size..."
|
||||||
OUTPUT_LORA_MERGED=$(./llama-cli -m $MODELS_REPO/$model_name/hidden_size=$hidden_size/base/Base-F32-lora-merged.gguf \
|
OUTPUT_LORA_MERGED=$(./llama-cli -no-cnv -m $MODELS_REPO/$model_name/hidden_size=$hidden_size/base/Base-F32-lora-merged.gguf \
|
||||||
-p "$EXPECTED_LORA_FIRST_WORD" -n 50 --seed 42 --temp 0)
|
-p "$EXPECTED_LORA_FIRST_WORD" -n 50 --seed 42 --temp 0)
|
||||||
|
|
||||||
# Remove any initial white space
|
# Remove any initial white space
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue