This commit is contained in:
Nick Walton
2019-11-18 18:29:38 -07:00
parent 8a049cef10
commit 55da36426a
3 changed files with 61 additions and 4 deletions
+2 -2
View File
@@ -1,10 +1,10 @@
MODEL_DIRECTORY=aidungeon/generator/gpt2/models/model_v2
MODEL_DIRECTORY=aidungeon/generator/gpt2/models/model_v1
if [ -d "$MODEL_DIRECTORY" ]; then
echo "AIDungeon2 is already installed"
else
echo "Downloading AIDungeon2 Model"
gsutil -m cp -r gs://aidungeon2model/model_v2 ./generator/gpt2/models
gsutil -m cp -r gs://aidungeon2model/model_v1 ./generator/gpt2/models
pip install -r requirements.txt > /dev/null
fi