Skip to content

Commit

Permalink
Add seamless text-to-speech and text-to-text translation with updated…
Browse files Browse the repository at this point in the history
… dependencies

- Create `seamless_v2.py` for new seamless text-to-speech and text-to-text translation tasks.
- Update Dockerfile to use CUDA 12.4.1 and Ubuntu 22.04.
- Upgrade dependencies in `requirements.txt`.
- Modify `model-values.yaml` to update image reference, GPU, and memory limits.
- Simplify `seamless_asr.py` for seamless ASR only (as huggingface expects a different model class for asr vs other tasks)
- Fix `run-dev.sh` to enable NVIDIA runtime and GPU usage.
  • Loading branch information
devxpy committed Aug 12, 2024
1 parent a5e5fb1 commit 73ff9c5
Show file tree
Hide file tree
Showing 8 changed files with 121 additions and 556 deletions.
5 changes: 3 additions & 2 deletions chart/model-values.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -195,9 +195,10 @@ deployments:
bark
- name: "common-seamless"
image: "us-docker.pkg.dev/dara-c1b52/cloudbuild/gooey-gpu/common@sha256:5fb0ffa128cbdda86747fedf5ef68e9df8256735d8535149c6fffa41a3749883"
image: "crgooeyprodwestus1.azurecr.io/gooey-gpu-common:3"
limits:
memory: "19Gi"
gpu: "10Gi"
memory: "28Gi" # (220 / 80) * 10
cpu: "1"
env:
IMPORTS: |-
Expand Down
2 changes: 1 addition & 1 deletion common/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM nvidia/cuda:12.2.0-devel-ubuntu20.04
FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04

ARG DEBIAN_FRONTEND=noninteractive

Expand Down
137 changes: 0 additions & 137 deletions common/deepfloyd.py

This file was deleted.

Loading

0 comments on commit 73ff9c5

Please sign in to comment.