Upload 3 files

Dockerfile

@@ -1,9 +1,11 @@
-# Base image with PyTorch + CUDA 12.1 runtime
-FROM pytorch/pytorch:2.5.1-cuda12.1-cudnn9-runtime
+# Python 3.10 base (works well with librosa 0.9.2 + numba 0.56.4)
+FROM pytorch/pytorch:2.3.1-cuda11.8-cudnn8-runtime
 
 ENV PYTHONUNBUFFERED=1 \
     PIP_DISABLE_PIP_VERSION_CHECK=1 \
-    PIP_PREFER_BINARY=1
+    PIP_PREFER_BINARY=1 \
+    NUMBA_CACHE_DIR=/tmp/numba_cache
+    # NUMBA_DISABLE_JIT=1   # ← 若仍报 numba JIT 缓存错误，可取消注释完全禁用 JIT（更稳，稍慢）
 
 WORKDIR /app
 
@@ -12,15 +14,17 @@ RUN apt-get update && \
     apt-get install -y --no-install-recommends \
       ffmpeg libsox-dev git \
       build-essential cmake ninja-build pkg-config && \
-    rm -rf /var/lib/apt/lists/*
+    rm -rf /var/lib/apt/lists/* && \
+    mkdir -p /tmp/numba_cache && chmod -R 777 /tmp/numba_cache
 
-# Get GPT-SoVITS source
+# Clone GPT-SoVITS
 RUN git clone --depth 1 https://github.com/RVC-Boss/GPT-SoVITS.git /app
 
 # Python deps
 RUN pip install --upgrade pip && \
-    pip install --no-deps --no-cache-dir -r /app/extra-req.txt && \
     pip install --no-cache-dir -r /app/requirements.txt && \
+    # 为防 requirements 中把 librosa/numba 升级，这里再次强制 pin 到兼容版本
+    pip install --no-cache-dir --force-reinstall --no-deps librosa==0.9.2 numba==0.56.4 && \
     pip install --no-cache-dir fastapi uvicorn soundfile huggingface_hub ffmpeg-python
 
 # Pre-download essentials

README.md

@@ -1,5 +1,5 @@
 ---
-title: GPT-SoVITS API (v2 ProPlus) for AstrBot
+title: GPT-SoVITS API (v2 ProPlus) for AstrBot (Py3.10 fix)
 emoji: 🗣️
 colorFrom: yellow
 colorTo: red
@@ -8,9 +8,18 @@ app_port: 7860
 license: mit
 ---
 
-# GPT‑SoVITS v2 ProPlus — REST API for AstrBot (Docker Space)
+# GPT‑SoVITS v2 ProPlus — REST API for AstrBot (Python 3.10 base)
 
-## 放置你的文件（统一命名，避免中文/空格/加号）
+> 该版本改用 **Python 3.10** 的 PyTorch 官方镜像（`pytorch/pytorch:2.3.1-cuda11.8-cudnn8-runtime`），
+> 并在 **安装完仓库 requirements 之后** 强制重装 `librosa==0.9.2`、`numba==0.56.4`，
+> 再结合 `NUMBA_CACHE_DIR=/tmp/numba_cache` 来规避你遇到的 `numba/librosa` 缓存错误。
+
+## 放置你的文件
 - GPT 权重 → `weights/shantianliang_proplus_e32.ckpt`
 - SoVITS 权重 → `weights/shantianliang_proplus_e8_s192.pth`
-- 参考音频 → `reference_audio/ref_shantianliang_1.wav`（当前是 1 秒静音占位，请替换）
+- 参考音频 → `reference_audio/ref_shantianliang_1.wav`（1 秒静音占位，建议替换）
+
+## 启动后测试（举例）
+- 切 SoVITS：`/set_sovits_weights?weights_path=/app/pretrained_models/shantianliang/shantianliang_proplus_e8_s192.pth`
+- 切 GPT：`/set_gpt_weights?weights_path=/app/pretrained_models/shantianliang/shantianliang_proplus_e32.ckpt`
+- 合成：`/tts`（POST JSON，支持流式/整段）