Fix indentation weirdness in GPT-2 example.

2019-04-22 02:20:22 +09:00
parent 68a889ee43
commit 14b1f719f4
1 changed files with 17 additions and 19 deletions
--- a/examples/run_gpt2.py
+++ b/examples/run_gpt2.py
@@ -107,7 +107,7 @@ def run_model():
                    print("=" * 40 + " SAMPLE " + str(generated) + " " + "=" * 40)
                    print(text)
            print("=" * 80)
-      if args.unconditional:
+        else:
            generated = 0
            for _ in range(args.nsamples // args.batch_size):
                out = sample_sequence(
@@ -124,8 +124,6 @@ def run_model():
                    print("=" * 40 + " SAMPLE " + str(generated) + " " + "=" * 40)
                    print(text)
            print("=" * 80)
-          if args.unconditional:
-              break

 if __name__ == '__main__':
    run_model()