From 207b3b7b11f09052b33f49372a09069c1d35b686 Mon Sep 17 00:00:00 2001 From: Nick Walton Date: Mon, 18 Nov 2019 22:30:53 -0700 Subject: [PATCH] update --- generator/gpt2/gpt2_generator.py | 2 +- install.sh | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index 9829f04..1a07568 100644 --- a/generator/gpt2/gpt2_generator.py +++ b/generator/gpt2/gpt2_generator.py @@ -17,7 +17,7 @@ class GPT2Generator: self.top_k = top_k self.top_p = top_p - self.model_name = "model_v3" + self.model_name = "model_v4" self.model_dir = "generator/gpt2/models" self.checkpoint_path = os.path.join(self.model_dir, self.model_name) diff --git a/install.sh b/install.sh index 6a9e27e..d8546db 100755 --- a/install.sh +++ b/install.sh @@ -1,10 +1,10 @@ -MODEL_DIRECTORY=aidungeon/generator/gpt2/models/model_v3 +MODEL_DIRECTORY=aidungeon/generator/gpt2/models/model_v4 if [ -d "$MODEL_DIRECTORY" ]; then echo "AIDungeon2 is already installed" else echo "Downloading AIDungeon2 Model" - gsutil -m cp -r gs://aidungeon2model/model_v3 ./generator/gpt2/models + gsutil -m cp -r gs://aidungeon2model/model_v4 ./generator/gpt2/models pip install -r requirements.txt > /dev/null fi