Update README.md

README.md

@@ -29,4 +29,6 @@ To generate text using the model:
 
 Training of the model is performed using the following loss computatation:
 
-    hidden_state.logits, hidden_state.loss = model(input_ids=input["input_ids"],attention_mask=input["attention_mask"],labels=input["labels"])
+    hidden_state.logits, hidden_state.loss = model(input_ids=input["input_ids"],
+                                                  attention_mask=input["attention_mask"],
+                                                  labels=input["labels"])