From d3bdb6e3ac433423b925fcd170b00188081cc50d Mon Sep 17 00:00:00 2001
From: Nick Walton <nickwalton00@gmail.com>
Date: Mon, 18 Nov 2019 12:10:45 -0700
Subject: [PATCH] update generator

---
 generator/gpt2/gpt2_generator.py | 3 ++-
 install_model.sh                 | 0
 2 files changed, 2 insertions(+), 1 deletion(-)
 mode change 100644 => 100755 install_model.sh

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