Update README.md

README.md

@@ -428,7 +428,7 @@ datasets:
 
 This model has the safetensors weights for the [Madlad-400](https://github.com/google-research/google-research/tree/master/madlad_400) 8B param **language model**.
 
-The Python code to run inference is not ready yet.
+The HF transformers code to run inference is not ready yet. The [original implementation](https://github.com/google/flaxformer/blob/ea17eb012a1d340ddff017b7a534c2162aaec34c/flaxformer/architectures/t5/t5_architecture.py#L1484) is in JAX/Flaxformer.
 
 The model architecture is the same as [Palm 8B](https://arxiv.org/pdf/2204.02311.pdf).