Update README.md

README.md

@@ -16,4 +16,4 @@ pipeline_tag: text-generation
 
 If you want to try how to use it here is a file of how to use it in [test_gen.py](https://github.com/F4k3r22/LLaDA-from-scratch/blob/main/test_gen.py) Or using this Google [Colab](https://colab.research.google.com/drive/1jPIPu9qHEFMkANzUEkeOxUW6hS3DeVwd?usp=sharing) notebook
 
-### For those who want to train and get the correct format to be able to load it with `transformers`, everything needed is in [`pre_train.py`](https://github.com/F4k3r22/LLaDA-from-scratch/blob/main/pre_train.py) of the project repo
+For those who want to train and get the correct format to be able to load it with `transformers`, everything needed is in [`pre_train.py`](https://github.com/F4k3r22/LLaDA-from-scratch/blob/main/pre_train.py) of the project repo