This commit is contained in:
Nick Walton
2019-11-18 17:17:10 -07:00
parent 55c22f30e1
commit c7892d1bcd
3 changed files with 4 additions and 5 deletions
+2 -3
View File
@@ -1,8 +1,7 @@
from story.story_manager import *
from generator.gpt2.gpt2_generator import *
CRED_FILE = "./AI-Adventure-2bb65e3a4e2f.json"
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
def console_print(text, width=75):
last_newline = 0
+1 -1
View File
@@ -86,7 +86,7 @@ class GPT2Generator:
def generate(self, prompt, options=None, seed=1):
debug_print=True
debug_print = False
prefix = self.prompt_replace(prompt)
if debug_print:
+1 -1
View File
@@ -6,5 +6,5 @@ if [ -d "$MODEL_DIRECTORY" ]; then
else
echo "Installing AIDungeon"
gsutil -m cp -r gs://aidungeon2model/model_v1 ./generator/gpt2/models
pip install -r requirements.txt
pip install -r requirements.txt > /dev/null
fi