Update README.md

README.md

@@ -22,7 +22,8 @@ tokens = tokenizer.encode_plus(prompt, truncation=False)["input_ids"]
 tokens = torch.tensor(tokens)[None,].to(device)
 with torch.no_grad():
     input_length = len(tokens[0])
-    out_ids = model.generate(input_ids=tokens, max_length=input_length+200, temperature=1.0, top_p=0.95, eos_token_id=tokenizer.eos_token_id, cg=True, top_k=15)
+    out_ids = model.generate(input_ids=tokens, max_length=input_length+200, temperature=1.0,
+                             top_p=0.95, eos_token_id=tokenizer.eos_token_id, cg=True, top_k=15)
     out_ids = out_ids[0][input_length:].cpu().numpy()
     out_text = tokenizer.decode(out_ids.tolist())
     print(out_text)