Files
Clover-Edition/install.sh
T
Nick Walton 436c421ff8 Develop (#12)
* adds saving and loading

* fixed some cmd issues

* fixed some cmd issues

* fixed capitalization issue

* smallupdate

* update

* update

* dev changes

* more testing

* more dev fixes

* dragon story

* dragon story

* adding ai dm mode

* dm fix

* dm

* dm

* Gcloudstoragestuff

* fixed loading issues

* upped memory

* upgrade to new model

* turn on debug

* fixed repetition issue

* turned off debug print

* upped dm temp
2019-12-02 08:36:43 -07:00

27 lines
1.0 KiB
Bash
Executable File

MODELS_DIRECTORY=generator/gpt2/models
MODEL_VERSION=model_v5
MODEL_NAME=model-550
DOWNLOAD_URL=https://students.cs.byu.edu/~nickwalt
if [ -d "${MODELS_DIRECTORY}/${MODEL_VERSION}" ]; then
echo "AIDungeon2 is already installed"
else
echo "Downloading AIDungeon2 Model... (this may take a few minutes)"
cd ${MODELS_DIRECTORY}
mkdir ${MODEL_VERSION}
cd ${MODEL_VERSION}
apt-get install aria2 > /dev/null
aria2c -x 16 -s 32 "${DOWNLOAD_URL}/${MODEL_VERSION}/${MODEL_NAME}.data-00000-of-00001"
wget "${DOWNLOAD_URL}/${MODEL_VERSION}/checkpoint" > /dev/null
wget "${DOWNLOAD_URL}/${MODEL_VERSION}/encoder.json" > /dev/null
wget "${DOWNLOAD_URL}/${MODEL_VERSION}/hparams.json" > /dev/null
wget "${DOWNLOAD_URL}/${MODEL_VERSION}/${MODEL_NAME}.index" > /dev/null
wget "${DOWNLOAD_URL}/${MODEL_VERSION}/${MODEL_NAME}.meta" > /dev/null
wget "${DOWNLOAD_URL}/${MODEL_VERSION}/vocab.bpe" > /dev/null
echo "Download Complete!"
cd ../../../..
pip install -r requirements.txt > /dev/null
fi