Update README.md

README.md

@@ -23,4 +23,4 @@ Example of the results it gives:
 
 ![image/png](lladaout.PNG)
 
-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_trainv2.py`](https://github.com/F4k3r22/LLaDA-from-scratch/blob/main/pre_trainv2.py) of the project repo