Merge pull request graykode#5 from mturnshek/master

change temperature argument type int -> float

Author: graykode

main.py

@@ -23,7 +23,7 @@ def text_generator(state_dict):
     parser.add_argument('--unconditional', action='store_true', help='If true, unconditional generation.')
     parser.add_argument("--batch_size", type=int, default=-1)
     parser.add_argument("--length", type=int, default=-1)
-    parser.add_argument("--temperature", type=int, default=0.7)
+    parser.add_argument("--temperature", type=float, default=0.7)
     parser.add_argument("--top_k", type=int, default=40)
     args = parser.parse_args()
 
@@ -79,4 +79,4 @@ def text_generator(state_dict):
         text_generator(state_dict)
     else:
         print('Please download gpt2-pytorch_model.bin')
-        sys.exit()
+        sys.exit()