Skip to content

kokoro TTS Supported axera backend #3281

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Abandon-ht wants to merge 10 commits into
base: master
Choose a base branch
from
Open

kokoro TTS Supported axera backend #3281

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
0aa35ef
kokoro TTS Supported axcl backend
Mar 9, 2026
0463121
fix offline tts kokoro axcl impl
Mar 9, 2026
1b940fa
kokoro TTS Supported axera backend
Mar 11, 2026
dd29306
Read metadata from model4 for AXCL and AXERA
Mar 11, 2026
4cdc81f
use comma-separated model paths instead of directory
Mar 11, 2026
e6fb94b
Refactor input preparation to support variable repeat counts in Offli…
Apr 1, 2026
f859ba9
Implement feature X to enhance user experience and optimize performance
Apr 3, 2026
b4b4515
Implement feature X to enhance user experience and optimize performance
Apr 3, 2026
3dd5b29
Add BSP SDK and model files to .gitignore
Apr 7, 2026
6823405
Enhance Axcl and Axera models with new tensor shape handling and upda…
Apr 7, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -177,3 +177,9 @@ sherpa-onnx-fire-red-asr2-ctc-zh_en-int8-2026-02-25
non-streaming-fire-red-asr-ctc-decode-files
sherpa-onnx-moonshine-*-quantized-2026-02-27
sherpa-onnx-supertonic-tts-int8-2026-03-06

ax620e_bsp_sdk-2.0.0_P7
ax650n_bsp_sdk-1.45.0_p39
aarch64-linux-gnu-g*
kokoro-multi-lang-v1_1
kokoro-v1_1-zh.pth
10 changes: 10 additions & 0 deletions sherpa-onnx/csrc/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -211,6 +211,11 @@ if(SHERPA_ONNX_ENABLE_AXERA)
./axera/offline-sense-voice-model-axera.cc
./axera/utils.cc
)
if(SHERPA_ONNX_ENABLE_TTS)
list(APPEND sources
./axera/offline-tts-kokoro-model-axera.cc
)
endif()
endif()

if(SHERPA_ONNX_ENABLE_AXCL)
Expand All @@ -223,6 +228,11 @@ if(SHERPA_ONNX_ENABLE_AXCL)
./axcl/offline-sense-voice-model-axcl.cc
./axcl/utils.cc
)
if(SHERPA_ONNX_ENABLE_TTS)
list(APPEND sources
./axcl/offline-tts-kokoro-model-axcl.cc
)
endif()
endif()

if(SHERPA_ONNX_ENABLE_RKNN OR SHERPA_ONNX_ENABLE_ASCEND_NPU OR SHERPA_ONNX_ENABLE_QNN OR SHERPA_ONNX_ENABLE_AXERA OR SHERPA_ONNX_ENABLE_AXCL)
Expand Down
5 changes: 5 additions & 0 deletions sherpa-onnx/csrc/axcl/axcl-model.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -429,6 +429,11 @@ bool AxclModel::SetInputTensorData(const std::string &name, const int32_t *p,
return impl_->SetInputTensorData(name, p, n);
}

bool AxclModel::SetInputTensorData(const std::string &name, const uint8_t *p,
int32_t n) const {
return impl_->SetInputTensorData(name, p, n);
}

std::vector<float> AxclModel::GetOutputTensorData(
const std::string &name) const {
return impl_->GetOutputTensorData(name);
Expand Down
3 changes: 3 additions & 0 deletions sherpa-onnx/csrc/axcl/axcl-model.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,9 @@ class AxclModel {

bool SetInputTensorData(const std::string &name, const int32_t *p,
int32_t n) const;

bool SetInputTensorData(const std::string &name, const uint8_t *p,
int32_t n) const;

std::vector<float> GetOutputTensorData(const std::string &name) const;

Expand Down
Loading