Update README.md

README.md

@@ -62,10 +62,13 @@ tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
 
 if tokenizer.pad_token is None:
     tokenizer.add_special_tokens({'pad_token': '[PAD]'})
+    
 text = "Capital of USA is"
+
 device = "cuda:0"
 
 inputs = tokenizer(text, return_tensors="pt").to(device)
 
 outputs = model.generate(**inputs, pad_token_id= tokenizer.eos_token_id, max_new_tokens=20)
+
 print(tokenizer.decode(outputs[0], skip_special_tokens=True))