From 5ae8b94e3a84f25813eccbb3f9f25896fd8d72f4 Mon Sep 17 00:00:00 2001 From: Nick Walton Date: Mon, 18 Nov 2019 12:15:27 -0700 Subject: [PATCH] fixed new generator --- generator/gpt2/gpt2_generator.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index 35d62c6..5cff3da 100644 --- a/generator/gpt2/gpt2_generator.py +++ b/generator/gpt2/gpt2_generator.py @@ -35,8 +35,10 @@ class GPT2Generator: hparams.override_from_dict(json.load(f)) seed = 20 - graph = tf.Graph() - self.sess = tf.Session(graph=graph) + config = tf.compat.v1.ConfigProto() + config.gpu_options.allow_growth = True + self.sess = tf.compat.v1.Session(config=config) + self.context = tf.placeholder(tf.int32, [self.batch_size, None]) np.random.seed(seed) tf.set_random_seed(seed) @@ -95,9 +97,7 @@ class GPT2Generator: print("******DEBUG******") print("Prompt is: ", repr(prefix)) - raw_text = input("Model prompt >>> ") - - context_tokens = self.enc.encode(raw_text) + context_tokens = self.enc.encode(prompt) generated = 0 for _ in range(self.samples // self.batch_size): out = self.sess.run(self.output, feed_dict={