28 lines
757 B
Bash
Executable file
28 lines
757 B
Bash
Executable file
#!/usr/bin/env bash
|
|
|
|
set -e
|
|
|
|
: ${MODEL_DIR:="pretrained_models/fastpitch"}
|
|
MODEL_ZIP="nvidia_fastpitch_210824.zip"
|
|
MODEL="nvidia_fastpitch_210824.pt"
|
|
MODEL_URL="https://api.ngc.nvidia.com/v2/models/nvidia/fastpitch_pyt_amp_ckpt_v1_1/versions/21.05.0/zip"
|
|
|
|
mkdir -p "$MODEL_DIR"
|
|
|
|
if [ ! -f "${MODEL_DIR}/${MODEL_ZIP}" ]; then
|
|
echo "Downloading ${MODEL_ZIP} ..."
|
|
wget -qO ${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
|