Compare commits
3 Commits
main
...
766ca9d278
| Author | SHA1 | Date | |
|---|---|---|---|
| 766ca9d278 | |||
| 9a017df4ca | |||
| fe3c77ff4f |
@@ -4,6 +4,7 @@ on:
|
||||
push:
|
||||
branches:
|
||||
- main
|
||||
- dev
|
||||
|
||||
jobs:
|
||||
build:
|
||||
@@ -22,14 +23,21 @@ jobs:
|
||||
username: ${{ secrets.REGISTRY_USERNAME }}
|
||||
password: ${{ secrets.REGISTRY_TOKEN }}
|
||||
|
||||
- name: Set tags
|
||||
id: tags
|
||||
run: |
|
||||
if [ "${{ gitea.ref_name }}" = "main" ]; then
|
||||
echo "tags=git.sdgarren.com/scott/rocm-chatterbox-whisper:latest,git.sdgarren.com/scott/rocm-chatterbox-whisper:${{ gitea.sha }}" >> $GITHUB_OUTPUT
|
||||
else
|
||||
echo "tags=git.sdgarren.com/scott/rocm-chatterbox-whisper:${{ gitea.ref_name }},git.sdgarren.com/scott/rocm-chatterbox-whisper:${{ gitea.sha }}" >> $GITHUB_OUTPUT
|
||||
fi
|
||||
|
||||
- name: Build and Push
|
||||
uses: docker/build-push-action@v6
|
||||
with:
|
||||
context: .
|
||||
file: Dockerfile.rocm
|
||||
push: true
|
||||
tags: |
|
||||
git.sdgarren.com/scott/rocm-chatterbox-whisper:latest
|
||||
git.sdgarren.com/scott/rocm-chatterbox-whisper:${{ gitea.sha }}
|
||||
tags: ${{ steps.tags.outputs.tags }}
|
||||
cache-from: type=registry,ref=git.sdgarren.com/scott/rocm-chatterbox-whisper:latest
|
||||
cache-to: type=inline
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
FROM rocm/dev-ubuntu-22.04:6.1
|
||||
FROM rocm/dev-ubuntu-22.04:7.2
|
||||
|
||||
ENV DEBIAN_FRONTEND=noninteractive \
|
||||
PYTHONDONTWRITEBYTECODE=1 \
|
||||
@@ -7,27 +7,37 @@ ENV DEBIAN_FRONTEND=noninteractive \
|
||||
PIP_NO_CACHE_DIR=1
|
||||
|
||||
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||
python3 \
|
||||
python3-pip \
|
||||
python3-dev \
|
||||
software-properties-common \
|
||||
&& add-apt-repository ppa:deadsnakes/ppa \
|
||||
&& apt-get update && apt-get install -y --no-install-recommends \
|
||||
python3.11 \
|
||||
python3.11-dev \
|
||||
python3.11-distutils \
|
||||
git \
|
||||
ffmpeg \
|
||||
libsndfile1 \
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
# Install pip for Python 3.11
|
||||
RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3.11
|
||||
|
||||
# Make python3.11 the default python3
|
||||
RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.11 1 \
|
||||
&& update-alternatives --install /usr/bin/python python /usr/bin/python3.11 1
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
# Step 1: Install ROCm-compatible PyTorch stack first.
|
||||
# This must happen before anything else to prevent pip from pulling CPU wheels.
|
||||
COPY requirements-rocm-init.txt .
|
||||
RUN pip3 install -r requirements-rocm-init.txt
|
||||
RUN python3 -m pip install -r requirements-rocm-init.txt
|
||||
|
||||
# Step 2: Install remaining dependencies (pinned to avoid overwriting torch).
|
||||
COPY requirements-rocm.txt .
|
||||
RUN pip3 install -r requirements-rocm.txt
|
||||
RUN python3 -m pip install -r requirements-rocm.txt
|
||||
|
||||
# Step 3: Install chatterbox with --no-deps so pip cannot replace ROCm torch.
|
||||
RUN pip3 install --no-deps chatterbox-tts
|
||||
RUN python3 -m pip install --no-deps chatterbox-tts
|
||||
|
||||
# Application source
|
||||
COPY engine.py config.py wyoming_handler.py wyoming_voices.py main.py ./
|
||||
|
||||
@@ -13,11 +13,11 @@ tts_engine:
|
||||
generation_defaults:
|
||||
# Turbo model: uses temperature only (exaggeration/cfg_weight ignored)
|
||||
# Standard model: uses exaggeration and cfg_weight (temperature ignored)
|
||||
temperature: 0.8
|
||||
temperature: 0.9
|
||||
exaggeration: 0.5
|
||||
cfg_weight: 0.5
|
||||
# seed: 0 = random each call, >0 = reproducible output
|
||||
seed: 0
|
||||
seed: 1960
|
||||
|
||||
wyoming:
|
||||
host: "0.0.0.0"
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
--index-url https://download.pytorch.org/whl/rocm6.1
|
||||
torch==2.5.1
|
||||
torchaudio==2.5.1
|
||||
torchvision==0.20.1
|
||||
pytorch_triton_rocm==3.1.0
|
||||
--index-url https://download.pytorch.org/whl/rocm7.2
|
||||
torch==2.11.0
|
||||
torchaudio==2.11.0
|
||||
torchvision==0.26.0
|
||||
pytorch_triton_rocm==3.3.0
|
||||
|
||||
@@ -5,9 +5,9 @@ librosa==0.11.0
|
||||
pyloudnorm
|
||||
|
||||
# ML dependencies (pinned to match chatterbox without overwriting ROCm torch)
|
||||
transformers==4.46.3
|
||||
diffusers==0.29.0
|
||||
safetensors>=0.4.1
|
||||
transformers>=4.50.0
|
||||
diffusers>=0.32.0
|
||||
safetensors>=0.4.5
|
||||
huggingface-hub
|
||||
omegaconf
|
||||
|
||||
|
||||
Reference in New Issue
Block a user