mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-09-09 11:13:26 +08:00
update
This commit is contained in:
+2
-3
@@ -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
|
||||
|
||||
@@ -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
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user