From 14b1f719f4c9860a3df8663e7e4ecac25182034a Mon Sep 17 00:00:00 2001
From: Sangwhan Moon <sangwhan@iki.fi>
Date: Mon, 22 Apr 2019 02:20:22 +0900
Subject: [PATCH] Fix indentation weirdness in GPT-2 example.

---
 examples/run_gpt2.py | 36 +++++++++++++++++-------------------
 1 file changed, 17 insertions(+), 19 deletions(-)

diff --git a/examples/run_gpt2.py b/examples/run_gpt2.py
index 9e9e2b255b..8f8208bbcd 100644
--- a/examples/run_gpt2.py
+++ b/examples/run_gpt2.py
@@ -107,25 +107,23 @@ def run_model():
                     print("=" * 40 + " SAMPLE " + str(generated) + " " + "=" * 40)
                     print(text)
             print("=" * 80)
-      if args.unconditional:
-          generated = 0
-          for _ in range(args.nsamples // args.batch_size):
-              out = sample_sequence(
-                  model=model, length=args.length,
-                  context=None,
-                  start_token=enc.encoder['<|endoftext|>'],
-                  batch_size=args.batch_size,
-                  temperature=args.temperature, top_k=args.top_k, device=device
-              )
-              out = out[:,1:].tolist()
-              for i in range(args.batch_size):
-                  generated += 1
-                  text = enc.decode(out[i])
-                  print("=" * 40 + " SAMPLE " + str(generated) + " " + "=" * 40)
-                  print(text)
-          print("=" * 80)
-          if args.unconditional:
-              break
+        else:
+            generated = 0
+            for _ in range(args.nsamples // args.batch_size):
+                out = sample_sequence(
+                    model=model, length=args.length,
+                    context=None,
+                    start_token=enc.encoder['<|endoftext|>'],
+                    batch_size=args.batch_size,
+                    temperature=args.temperature, top_k=args.top_k, device=device
+                )
+                out = out[:,1:].tolist()
+                for i in range(args.batch_size):
+                    generated += 1
+                    text = enc.decode(out[i])
+                    print("=" * 40 + " SAMPLE " + str(generated) + " " + "=" * 40)
+                    print(text)
+            print("=" * 80)
 
 if __name__ == '__main__':
     run_model()