diff --git a/install.sh b/install.sh index 9fff4a3..358bdf4 100755 --- a/install.sh +++ b/install.sh @@ -7,20 +7,11 @@ 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 "Downloading AIDungeon2 Model... (this may take a few minutes)" + gsutil -m cp -r gs://multiregionaidungeon2/model_v5 . echo "Download Complete!" - cd ../../../.. + cd ../../.. pip install -r requirements.txt > /dev/null fi