Skip to content

Commit 36c373b

Browse files
authored
feat(kokoro): add support for l4t devices (#6322)
Signed-off-by: Ettore Di Giacinto <[email protected]>
1 parent 6afcb93 commit 36c373b

File tree

3 files changed

+31
-0
lines changed

3 files changed

+31
-0
lines changed

.github/workflows/backend.yml

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -489,6 +489,18 @@ jobs:
489489
backend: "diffusers"
490490
dockerfile: "./backend/Dockerfile.python"
491491
context: "./backend"
492+
- build-type: 'l4t'
493+
cuda-major-version: "12"
494+
cuda-minor-version: "0"
495+
platforms: 'linux/arm64'
496+
tag-latest: 'auto'
497+
tag-suffix: '-gpu-nvidia-l4t-kokoro'
498+
runs-on: 'ubuntu-24.04-arm'
499+
base-image: "nvcr.io/nvidia/l4t-jetpack:r36.4.0"
500+
skip-drivers: 'true'
501+
backend: "kokoro"
502+
dockerfile: "./backend/Dockerfile.python"
503+
context: "./backend"
492504
# SYCL additional backends
493505
- build-type: 'intel'
494506
cuda-major-version: ""

backend/index.yaml

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -270,6 +270,7 @@
270270
nvidia: "cuda12-kokoro"
271271
intel: "intel-kokoro"
272272
amd: "rocm-kokoro"
273+
nvidia-l4t: "nvidia-l4t-kokoro"
273274
- &coqui
274275
urls:
275276
- https://github.com/idiap/coqui-ai-TTS
@@ -1049,6 +1050,7 @@
10491050
nvidia: "cuda12-kokoro-development"
10501051
intel: "intel-kokoro-development"
10511052
amd: "rocm-kokoro-development"
1053+
nvidia-l4t: "nvidia-l4t-kokoro-development"
10521054
- !!merge <<: *kokoro
10531055
name: "cuda11-kokoro-development"
10541056
uri: "quay.io/go-skynet/local-ai-backends:master-gpu-nvidia-cuda-11-kokoro"
@@ -1074,6 +1076,16 @@
10741076
uri: "quay.io/go-skynet/local-ai-backends:master-gpu-intel-kokoro"
10751077
mirrors:
10761078
- localai/localai-backends:master-gpu-intel-kokoro
1079+
- !!merge <<: *kokoro
1080+
name: "nvidia-l4t-kokoro"
1081+
uri: "quay.io/go-skynet/local-ai-backends:latest-gpu-nvidia-l4t-kokoro"
1082+
mirrors:
1083+
- localai/localai-backends:latest-gpu-nvidia-l4t-kokoro
1084+
- !!merge <<: *kokoro
1085+
name: "nvidia-l4t-kokoro-development"
1086+
uri: "quay.io/go-skynet/local-ai-backends:master-gpu-nvidia-l4t-kokoro"
1087+
mirrors:
1088+
- localai/localai-backends:master-gpu-nvidia-l4t-kokoro
10771089
- !!merge <<: *kokoro
10781090
name: "cuda11-kokoro"
10791091
uri: "quay.io/go-skynet/local-ai-backends:latest-gpu-nvidia-cuda-11-kokoro"
Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
--extra-index-url https://pypi.jetson-ai-lab.io/jp6/cu126/
2+
torch
3+
torchaudio
4+
transformers
5+
accelerate
6+
kokoro
7+
soundfile

0 commit comments

Comments
 (0)