File size: 887 Bytes
7cbe545
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
# Inflect-Micro-v2 TTS SDK for AX650 NPU3

Encoder-Decoder split VITS text-to-speech engine.

## Files
- `inflect_encoder.axmodel` β€” Text encoder (NPU)
- `inflect_decoder.axmodel` β€” Flow + HiFi-GAN decoder (NPU)  
- `model.pth` β€” Full PyTorch weights (CPU components: embedding, duration, flow params)
- `config.json` β€” Model configuration
- `tts_engine.py` β€” Main inference engine

## Usage
```python
from inflect_tts_sdk import InflectTTSEngine

engine = InflectTTSEngine()
engine.warmup()  # Initialize NPU sessions

sr, wav = engine.synthesize("Hello world!")
engine.save("Hello world!", "output.wav")
```

## Pipeline
1. Text β†’ Phonemes β†’ Tokens (CPU, Python frontend)
2. Embedding lookup (CPU)
3. **Encoder NPU**: embeddings β†’ m_p, logs_p, x_mask
4. Duration predictor + alignment (CPU)
5. **Decoder NPU**: z_p β†’ waveform

## Requirements
See `requirements.txt`.