Spaces:
Sleeping
Sleeping
File size: 2,161 Bytes
8146713 |
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 |
#!/usr/bin/env bash
set -e
MODEL_NAMES="$@"
[ -z "$MODEL_NAMES" ] && { echo "Usage: $0 [fastpitch|waveglow|hifigan|hifigan-finetuned-fastpitch]"; exit 1; }
function download_ngc_model() {
mkdir -p "$MODEL_DIR"
if [ ! -f "${MODEL_DIR}/${MODEL_ZIP}" ]; then
echo "Downloading ${MODEL_ZIP} ..."
wget --content-disposition -O ${MODEL_DIR}/${MODEL_ZIP} ${MODEL_URL} \
|| { echo "ERROR: Failed to download ${MODEL_ZIP} from NGC"; exit 1; }
fi
if [ ! -f "${MODEL_DIR}/${MODEL}" ]; then
echo "Extracting ${MODEL} ..."
unzip -qo ${MODEL_DIR}/${MODEL_ZIP} -d ${MODEL_DIR} \
|| { echo "ERROR: Failed to extract ${MODEL_ZIP}"; exit 1; }
echo "OK"
else
echo "${MODEL} already downloaded."
fi
}
for MODEL_NAME in $MODEL_NAMES
do
case $MODEL_NAME in
"fastpitch")
MODEL_DIR="pretrained_models/fastpitch"
MODEL_ZIP="fastpitch_pyt_fp32_ckpt_v1_1_21.05.0.zip"
MODEL="nvidia_fastpitch_210824.pt"
MODEL_URL="https://api.ngc.nvidia.com/v2/models/nvidia/fastpitch_pyt_fp32_ckpt_v1_1/versions/21.05.0/zip"
;;
"hifigan")
MODEL_DIR="pretrained_models/hifigan"
MODEL_ZIP="hifigan__pyt_ckpt_ds-ljs22khz_21.08.0_amp.zip"
MODEL="hifigan_gen_checkpoint_6500.pt"
MODEL_URL="https://api.ngc.nvidia.com/v2/models/nvidia/dle/hifigan__pyt_ckpt_ds-ljs22khz/versions/21.08.0_amp/zip"
;;
"hifigan-finetuned-fastpitch")
MODEL_DIR="pretrained_models/hifigan"
MODEL_ZIP="hifigan__pyt_ckpt_mode-finetune_ds-ljs22khz_21.08.0_amp.zip"
MODEL="hifigan_gen_checkpoint_10000_ft.pt"
MODEL_URL="https://api.ngc.nvidia.com/v2/models/nvidia/dle/hifigan__pyt_ckpt_mode-finetune_ds-ljs22khz/versions/21.08.0_amp/zip"
;;
"waveglow")
MODEL_DIR="pretrained_models/waveglow"
MODEL_ZIP="waveglow_ckpt_amp_256_20.01.0.zip"
MODEL="nvidia_waveglow256pyt_fp16.pt"
MODEL_URL="https://api.ngc.nvidia.com/v2/models/nvidia/waveglow_ckpt_amp_256/versions/20.01.0/zip"
;;
*)
echo "Unrecognized model: ${MODEL_NAME}"
exit 2
;;
esac
download_ngc_model "$MODEL_DIR" "$MODEL_ZIP" "$MODEL" "$MODEL_URL"
done
|