Update README.md

Author: conceptofmind

README.md

@@ -1,6 +1,8 @@
-## T5 - PyTorch
+## T5 - PyTorch (WIP)
 A PyTorch implementation of [Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer](https://arxiv.org/abs/1910.10683). You can find the official T5x repository by Google [here](https://github.com/google-research/t5x).
 
+There is a small bug with dimensionality which needs to be resolved.
+
 ## Acknowledgement
 
 Phil Wang (lucidrains) advised and provided review for this implementation. [Please be sure to follow and support his work](https://github.com/lucidrains?tab=repositories).
@@ -61,4 +63,4 @@ Transfer learning, where a model is first pre-trained on a data-rich task before
   
   copyright = {arXiv.org perpetual, non-exclusive license}
 }
-```
+```