diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index 8fe54b5..35d62c6 100644 --- a/generator/gpt2/gpt2_generator.py +++ b/generator/gpt2/gpt2_generator.py @@ -35,7 +35,8 @@ class GPT2Generator: hparams.override_from_dict(json.load(f)) seed = 20 - self.sess = tf.Session(graph=tf.Graph()) + graph = tf.Graph() + self.sess = tf.Session(graph=graph) self.context = tf.placeholder(tf.int32, [self.batch_size, None]) np.random.seed(seed) tf.set_random_seed(seed) diff --git a/install_model.sh b/install_model.sh old mode 100644 new mode 100755