yangrongzhao
Inflect-Micro-v2 TTS: AX650 NPU3, U16 AXMODEL
7cbe545
|
Raw
History Blame Contribute Delete
887 Bytes
# 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`.