From 66445fad8401e574e1035b0e14161724eb049af1 Mon Sep 17 00:00:00 2001
From: scott <scott@sdgarren.com>
Date: Mon, 6 Apr 2026 18:05:43 -0400
Subject: [PATCH] Revert "Switch to ROCm 6.4 + Python 3.11 + PyTorch 2.9.1"

This reverts commit eb6a39d2928bf4df26c291c77e367a787a72cf8d.
---
 Dockerfile.rocm            | 24 +++++++-----------------
 requirements-rocm-init.txt | 10 +++++-----
 2 files changed, 12 insertions(+), 22 deletions(-)

diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index 43812c4..30ba3b2 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -1,4 +1,4 @@
-FROM rocm/dev-ubuntu-22.04:6.4
+FROM rocm/dev-ubuntu-22.04:6.1
 
 ENV DEBIAN_FRONTEND=noninteractive \
     PYTHONDONTWRITEBYTECODE=1 \
@@ -7,37 +7,27 @@ ENV DEBIAN_FRONTEND=noninteractive \
     PIP_NO_CACHE_DIR=1
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    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 \
+    python3 \
+    python3-pip \
+    python3-dev \
     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 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).
 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.
-RUN python3 -m pip install --no-deps chatterbox-tts
+RUN pip3 install --no-deps chatterbox-tts
 
 # Application source
 COPY engine.py config.py wyoming_handler.py wyoming_voices.py main.py ./
diff --git a/requirements-rocm-init.txt b/requirements-rocm-init.txt
index ce4fabe..44a44f9 100644
--- a/requirements-rocm-init.txt
+++ b/requirements-rocm-init.txt
@@ -1,5 +1,5 @@
---index-url https://download.pytorch.org/whl/rocm6.4
-torch==2.9.1
-torchaudio==2.9.1
-torchvision==0.24.1
-pytorch_triton_rocm==3.5.1
+--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