From fd8a3556f08bbcfb9c4f3eadea6206751c1b1dd9 Mon Sep 17 00:00:00 2001
From: Benjamin Mann <8enmann@gmail.com>
Date: Mon, 8 Apr 2019 17:20:35 -0700
Subject: [PATCH] fix run_gpt2.py

---
 examples/run_gpt2.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/examples/run_gpt2.py b/examples/run_gpt2.py
index 0350747499..a30c6c6456 100644
--- a/examples/run_gpt2.py
+++ b/examples/run_gpt2.py
@@ -83,7 +83,8 @@ def run_model():
     elif args.length > model.config.n_ctx:
         raise ValueError("Can't get samples longer than window size: %s" % model.config.n_ctx)
 
-    while not args.unconditional:
+    while True:
+        context_tokens = []
         if not args.unconditional:
             raw_text = input("Model prompt >>> ")
             while not raw_text:
@@ -106,6 +107,8 @@ def run_model():
                 print("=" * 40 + " SAMPLE " + str(generated) + " " + "=" * 40)
                 print(text)
         print("=" * 80)
+        if args.unconditional:
+            break
 
 if __name__ == '__main__':
     run_model()