Update README.md

README.md

@@ -14,6 +14,6 @@ pipeline_tag: text-generation
 
 # DISCLAIMER: Ok, a 310M model has been trained in a test in only 200 steps, on a mini dataset of 40,960 tokens the model is not competent, it is only a test and is compatible with Transformers
 
-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)
+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