Update README.md

README.md

@@ -42,10 +42,17 @@ outputs = model(input_ids=input_ids, use_cache=True, past_key_values=past_key_va
 ######## Generate!
 with torch.autocast(**amp_settings), torch.no_grad():
     output_ids = model.generate(input_ids, max_new_tokens=20, use_cache=True, num_steps=32)
+
 print(tokenizer.decode(output_ids[0]))
 
 
 # with or without cache
 with torch.autocast(**amp_settings), torch.no_grad():
     output_ids = model.generate(input_ids, max_new_tokens=20, use_cache=False, num_steps=32)
+
 print(tokenizer.decode(output_ids[0]))
+
+# Both are supposed to print:
+
+# <|begin_text|>The capital of Westphalia is the city of Münster. The city is located in the north of the state and is
+```