diff --git a/console_play.py b/console_play.py index 89520f1..07eb74c 100644 --- a/console_play.py +++ b/console_play.py @@ -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 diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index aa2c20b..cf5ed85 100644 --- a/generator/gpt2/gpt2_generator.py +++ b/generator/gpt2/gpt2_generator.py @@ -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: diff --git a/install.sh b/install.sh index db2b7c4..4c67163 100755 --- a/install.sh +++ b/install.sh @@ -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