Update README.md

README.md

@@ -8,7 +8,7 @@ language:
 pipeline_tag: text-to-speech
 ---
 
-# F5 Hindi 24KHz Model
+# F5-TTS Hindi 24KHz Model
 
 This is a Hindi Text-to-Speech model trained from scratch using the [F5 architecure](https://arxiv.org/abs/2410.06885).
 
@@ -30,6 +30,8 @@ Clone the following github repo and refer to the README: https://github.com/rumo
 
 The model was trained on 8x A100 40GB GPUs for close to a week. We would like to thank [CDAC](https://cdac.in/) for providing the compute resources.
 
+We used the "small" configuration(151M parameter) model for training according to the F5 paper.
+
 ### Training Data
 
 We used the Hindi subsets of [IndicTTS](https://www.tsdconference.org/tsd2016/download/cbblr16-850.pdf) and [IndicVoices-R](https://arxiv.org/pdf/2409.05356) datasets for training this model.