close code block in README

README.md

@@ -8700,6 +8700,7 @@ for i in tqdm(range(0, len(minicorpus_docs["input_ids"]), batch_size)):
         )
 
 dataset_embeddings = torch.cat(dataset_embeddings)
+```
 
 ## Running the second stage