From 07154dadb4fc5ce47e3d82dd33debb8e588039bd Mon Sep 17 00:00:00 2001
From: Abhi Sharma <18308855+SudoSharma@users.noreply.github.com>
Date: Tue, 16 Apr 2019 11:11:49 -0700
Subject: [PATCH] Fix indentation for unconditional generation

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

diff --git a/examples/run_gpt2.py b/examples/run_gpt2.py
index 61fbf9f323..9ac2b31961 100644
--- a/examples/run_gpt2.py
+++ b/examples/run_gpt2.py
@@ -107,25 +107,25 @@ 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
+      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
 
 if __name__ == '__main__':
     run_model()