diff --git a/console_play.py b/console_play.py index 78aec3d..c305d29 100644 --- a/console_play.py +++ b/console_play.py @@ -22,6 +22,8 @@ def play_unconstrained(): generator = GPT2Generator() prompt = get_story_start("knight") context = get_context("knight") + print("DEUBUG: Context is ", context) + print("DEBUG: Prompt is ", prompt) story_manager = UnconstrainedStoryManager(generator) print("Generating initial story.") story_manager.start_new_story(prompt, context=context) diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index cf5ed85..fe97679 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 = False + debug_print = True prefix = self.prompt_replace(prompt) if debug_print: diff --git a/install.sh b/install.sh index 4c67163..5fb1acb 100755 --- a/install.sh +++ b/install.sh @@ -1,10 +1,10 @@ MODEL_DIRECTORY=aidungeon/generator/gpt2/models/model_v1 if [ -d "$MODEL_DIRECTORY" ]; then - echo "AIDungeon is already installed" + echo "AIDungeon2 is already installed" else - echo "Installing AIDungeon" + echo "Downloading AIDungeon2 Model" gsutil -m cp -r gs://aidungeon2model/model_v1 ./generator/gpt2/models pip install -r requirements.txt > /dev/null fi