Skip to content

Commit fb37276

Browse files
committed
feat(vibevoice): add backend
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
1 parent 193a9fe commit fb37276

18 files changed

+956
-1
lines changed

Makefile

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -287,12 +287,14 @@ prepare-test-extra: protogen-python
287287
$(MAKE) -C backend/python/diffusers
288288
$(MAKE) -C backend/python/chatterbox
289289
$(MAKE) -C backend/python/vllm
290+
$(MAKE) -C backend/python/vibevoice
290291

291292
test-extra: prepare-test-extra
292293
$(MAKE) -C backend/python/transformers test
293294
$(MAKE) -C backend/python/diffusers test
294295
$(MAKE) -C backend/python/chatterbox test
295296
$(MAKE) -C backend/python/vllm test
297+
$(MAKE) -C backend/python/vibevoice test
296298

297299
DOCKER_IMAGE?=local-ai
298300
DOCKER_AIO_IMAGE?=local-ai-aio
@@ -389,6 +391,9 @@ backends/neutts: docker-build-neutts docker-save-neutts build
389391
backends/vllm: docker-build-vllm docker-save-vllm build
390392
./local-ai backends install "ocifile://$(abspath ./backend-images/vllm.tar)"
391393

394+
backends/vibevoice: docker-build-vibevoice docker-save-vibevoice build
395+
./local-ai backends install "ocifile://$(abspath ./backend-images/vibevoice.tar)"
396+
392397
build-darwin-python-backend: build
393398
bash ./scripts/build/python-darwin.sh
394399

@@ -445,6 +450,9 @@ docker-save-kitten-tts: backend-images
445450
docker-save-chatterbox: backend-images
446451
docker save local-ai-backend:chatterbox -o backend-images/chatterbox.tar
447452

453+
docker-save-vibevoice: backend-images
454+
docker save local-ai-backend:vibevoice -o backend-images/vibevoice.tar
455+
448456
docker-build-neutts:
449457
docker build --build-arg BUILD_TYPE=$(BUILD_TYPE) --build-arg BASE_IMAGE=$(BASE_IMAGE) -t local-ai-backend:neutts -f backend/Dockerfile.python --build-arg BACKEND=neutts ./backend
450458

@@ -523,10 +531,13 @@ docker-build-bark:
523531
docker-build-chatterbox:
524532
docker build --build-arg BUILD_TYPE=$(BUILD_TYPE) --build-arg BASE_IMAGE=$(BASE_IMAGE) -t local-ai-backend:chatterbox -f backend/Dockerfile.python --build-arg BACKEND=chatterbox ./backend
525533

534+
docker-build-vibevoice:
535+
docker build --progress=plain --build-arg BUILD_TYPE=$(BUILD_TYPE) --build-arg BASE_IMAGE=$(BASE_IMAGE) -t local-ai-backend:vibevoice -f backend/Dockerfile.python --build-arg BACKEND=vibevoice ./backend
536+
526537
docker-build-exllama2:
527538
docker build --build-arg BUILD_TYPE=$(BUILD_TYPE) --build-arg BASE_IMAGE=$(BASE_IMAGE) -t local-ai-backend:exllama2 -f backend/Dockerfile.python --build-arg BACKEND=exllama2 .
528539

529-
docker-build-backends: docker-build-llama-cpp docker-build-rerankers docker-build-vllm docker-build-transformers docker-build-diffusers docker-build-kokoro docker-build-faster-whisper docker-build-coqui docker-build-bark docker-build-chatterbox docker-build-exllama2
540+
docker-build-backends: docker-build-llama-cpp docker-build-rerankers docker-build-vllm docker-build-transformers docker-build-diffusers docker-build-kokoro docker-build-faster-whisper docker-build-coqui docker-build-bark docker-build-chatterbox docker-build-vibevoice docker-build-exllama2
530541

531542
########################################################
532543
### END Backends

backend/python/vibevoice/Makefile

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
.PHONY: vibevoice
2+
vibevoice:
3+
bash install.sh
4+
5+
.PHONY: run
6+
run: vibevoice
7+
@echo "Running vibevoice..."
8+
bash run.sh
9+
@echo "vibevoice run."
10+
11+
.PHONY: test
12+
test: vibevoice
13+
@echo "Testing vibevoice..."
14+
bash test.sh
15+
@echo "vibevoice tested."
16+
17+
.PHONY: protogen-clean
18+
protogen-clean:
19+
$(RM) backend_pb2_grpc.py backend_pb2.py
20+
21+
.PHONY: clean
22+
clean: protogen-clean
23+
rm -rf venv __pycache__

0 commit comments

Comments
 (0)