File size: 885 Bytes
7f959ae |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
---
tags:
- text-to-speech
- tts
- pytorch
language:
- ar
license: mit
---
# Multi-Speaker TTS Model
Trained Text-to-Speech model with the following specs:
## Model Details
- **Speakers:** 1
- **Mel Channels:** 80
- **Sample Rate:** 22050
- **Vocab Size:** 50000
## Training Stats
- **Final Mel Loss:** 5.3586
- **Validation Mel Loss:** 5.4784
- **Training Epochs:** 105
## Mel Normalization
- **Mean:** -4.8906
- **Std:** 4.9261
## Usage
```python
import torch
from transformers import AutoTokenizer
# Load model
checkpoint = torch.load('checkpoint_best.pt')
model = MultiSpeakerTTS(checkpoint['config'])
model.load_state_dict(checkpoint['model_state_dict'])
model.eval()
# Generate
tokenizer = AutoTokenizer.from_pretrained("miscovery/tokenizer_v3")
text = "Your text here"
tokens = tokenizer.encode(text)
# ... inference code ...
```
## Speaker Mapping
{
"mahmoud": 0
}
|