Compare commits
3 Commits
d517f730c2
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
| e7a03b9f0f | |||
| 66445fad84 | |||
| eb6a39d292 |
@@ -1,10 +1,7 @@
|
|||||||
name: Build ROCm Image
|
name: Build ROCm Image
|
||||||
|
|
||||||
on:
|
on:
|
||||||
push:
|
workflow_dispatch:
|
||||||
branches:
|
|
||||||
- main
|
|
||||||
- dev
|
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
build:
|
build:
|
||||||
@@ -23,21 +20,14 @@ jobs:
|
|||||||
username: ${{ secrets.REGISTRY_USERNAME }}
|
username: ${{ secrets.REGISTRY_USERNAME }}
|
||||||
password: ${{ secrets.REGISTRY_TOKEN }}
|
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
|
- name: Build and Push
|
||||||
uses: docker/build-push-action@v6
|
uses: docker/build-push-action@v6
|
||||||
with:
|
with:
|
||||||
context: .
|
context: .
|
||||||
file: Dockerfile.rocm
|
file: Dockerfile.rocm
|
||||||
push: true
|
push: true
|
||||||
tags: ${{ steps.tags.outputs.tags }}
|
tags: |
|
||||||
|
git.sdgarren.com/scott/rocm-chatterbox-whisper:latest
|
||||||
|
git.sdgarren.com/scott/rocm-chatterbox-whisper:${{ gitea.sha }}
|
||||||
cache-from: type=registry,ref=git.sdgarren.com/scott/rocm-chatterbox-whisper:latest
|
cache-from: type=registry,ref=git.sdgarren.com/scott/rocm-chatterbox-whisper:latest
|
||||||
cache-to: type=inline
|
cache-to: type=inline
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
FROM rocm/dev-ubuntu-22.04:7.2
|
FROM rocm/dev-ubuntu-22.04:6.1
|
||||||
|
|
||||||
ENV DEBIAN_FRONTEND=noninteractive \
|
ENV DEBIAN_FRONTEND=noninteractive \
|
||||||
PYTHONDONTWRITEBYTECODE=1 \
|
PYTHONDONTWRITEBYTECODE=1 \
|
||||||
@@ -7,37 +7,27 @@ ENV DEBIAN_FRONTEND=noninteractive \
|
|||||||
PIP_NO_CACHE_DIR=1
|
PIP_NO_CACHE_DIR=1
|
||||||
|
|
||||||
RUN apt-get update && apt-get install -y --no-install-recommends \
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||||
software-properties-common \
|
python3 \
|
||||||
&& add-apt-repository ppa:deadsnakes/ppa \
|
python3-pip \
|
||||||
&& apt-get update && apt-get install -y --no-install-recommends \
|
python3-dev \
|
||||||
python3.11 \
|
|
||||||
python3.11-dev \
|
|
||||||
python3.11-distutils \
|
|
||||||
git \
|
git \
|
||||||
ffmpeg \
|
ffmpeg \
|
||||||
libsndfile1 \
|
libsndfile1 \
|
||||||
&& rm -rf /var/lib/apt/lists/*
|
&& 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
|
WORKDIR /app
|
||||||
|
|
||||||
# Step 1: Install ROCm-compatible PyTorch stack first.
|
# Step 1: Install ROCm-compatible PyTorch stack first.
|
||||||
# This must happen before anything else to prevent pip from pulling CPU wheels.
|
# This must happen before anything else to prevent pip from pulling CPU wheels.
|
||||||
COPY requirements-rocm-init.txt .
|
COPY requirements-rocm-init.txt .
|
||||||
RUN python3 -m pip install -r requirements-rocm-init.txt
|
RUN pip3 install -r requirements-rocm-init.txt
|
||||||
|
|
||||||
# Step 2: Install remaining dependencies (pinned to avoid overwriting torch).
|
# Step 2: Install remaining dependencies (pinned to avoid overwriting torch).
|
||||||
COPY requirements-rocm.txt .
|
COPY requirements-rocm.txt .
|
||||||
RUN python3 -m pip install -r requirements-rocm.txt
|
RUN pip3 install -r requirements-rocm.txt
|
||||||
|
|
||||||
# Step 3: Install chatterbox with --no-deps so pip cannot replace ROCm torch.
|
# Step 3: Install chatterbox with --no-deps so pip cannot replace ROCm torch.
|
||||||
RUN python3 -m pip install --no-deps chatterbox-tts
|
RUN pip3 install --no-deps chatterbox-tts
|
||||||
|
|
||||||
# Application source
|
# Application source
|
||||||
COPY engine.py config.py wyoming_handler.py wyoming_voices.py main.py ./
|
COPY engine.py config.py wyoming_handler.py wyoming_voices.py main.py ./
|
||||||
|
|||||||
@@ -13,11 +13,11 @@ tts_engine:
|
|||||||
generation_defaults:
|
generation_defaults:
|
||||||
# Turbo model: uses temperature only (exaggeration/cfg_weight ignored)
|
# Turbo model: uses temperature only (exaggeration/cfg_weight ignored)
|
||||||
# Standard model: uses exaggeration and cfg_weight (temperature ignored)
|
# Standard model: uses exaggeration and cfg_weight (temperature ignored)
|
||||||
temperature: 0.9
|
temperature: 0.8
|
||||||
exaggeration: 0.5
|
exaggeration: 0.5
|
||||||
cfg_weight: 0.5
|
cfg_weight: 0.5
|
||||||
# seed: 0 = random each call, >0 = reproducible output
|
# seed: 0 = random each call, >0 = reproducible output
|
||||||
seed: 1960
|
seed: 0
|
||||||
|
|
||||||
wyoming:
|
wyoming:
|
||||||
host: "0.0.0.0"
|
host: "0.0.0.0"
|
||||||
|
|||||||
@@ -32,8 +32,6 @@ services:
|
|||||||
- HSA_OVERRIDE_GFX_VERSION=10.3.0
|
- HSA_OVERRIDE_GFX_VERSION=10.3.0
|
||||||
# Disable MIOpen's SQLite cache — avoids crashes writing benchmark results.
|
# Disable MIOpen's SQLite cache — avoids crashes writing benchmark results.
|
||||||
- MIOPEN_DISABLE_CACHE=1
|
- MIOPEN_DISABLE_CACHE=1
|
||||||
# Suppress MIOpen workspace=0 solver warnings (expected with ROCm/PyTorch, not actionable).
|
|
||||||
- MIOPEN_LOG_LEVEL=2
|
|
||||||
|
|
||||||
volumes:
|
volumes:
|
||||||
hf_cache:
|
hf_cache:
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
--index-url https://download.pytorch.org/whl/rocm7.2
|
--index-url https://download.pytorch.org/whl/rocm6.1
|
||||||
torch==2.11.0
|
torch==2.5.1
|
||||||
torchaudio==2.11.0
|
torchaudio==2.5.1
|
||||||
torchvision==0.26.0
|
torchvision==0.20.1
|
||||||
pytorch_triton_rocm==3.3.0
|
pytorch_triton_rocm==3.1.0
|
||||||
|
|||||||
@@ -5,9 +5,9 @@ librosa==0.11.0
|
|||||||
pyloudnorm
|
pyloudnorm
|
||||||
|
|
||||||
# ML dependencies (pinned to match chatterbox without overwriting ROCm torch)
|
# ML dependencies (pinned to match chatterbox without overwriting ROCm torch)
|
||||||
transformers>=4.50.0
|
transformers==4.46.3
|
||||||
diffusers>=0.32.0
|
diffusers==0.29.0
|
||||||
safetensors>=0.4.5
|
safetensors>=0.4.1
|
||||||
huggingface-hub
|
huggingface-hub
|
||||||
omegaconf
|
omegaconf
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user