Spaces:
Runtime error
Runtime error
File size: 3,007 Bytes
f6c6cdf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
# 使用官方 PyTorch
FROM pytorch/pytorch:2.3.1-cuda11.8-cudnn8-runtime
# 切换到 root 用户以安装系统依赖
USER root
# 设置环境变量
ENV PYTHONUNBUFFERED=1 \
PIP_DISABLE_PIP_VERSION_CHECK=1 \
PIP_PREFER_BINARY=1 \
NUMBA_CACHE_DIR=/tmp/numba_cache
# 设置工作目录
WORKDIR /app
# 安装系统依赖
# --- 在这里添加了 wget ---
RUN apt-get update && \
apt-get install -y --no-install-recommends \
wget ffmpeg libsox-dev git \
build-essential cmake ninja-build pkg-config && \
rm -rf /var/lib/apt/lists/* && \
mkdir -p /tmp/numba_cache && chmod -R 777 /tmp/numba_cache
# [双重
RUN mkdir -p /nltk_data && chmod 777 /nltk_data
# 克隆 GPT-SoVITS 仓库
RUN git clone --depth 1 https://github.com/RVC-Boss/GPT-SoVITS.git /app
# 安装 Python 依赖
RUN pip install --upgrade pip && \
pip install --no-cache-dir -r /app/requirements.txt && \
pip install --no-cache-dir --force-reinstall numpy==1.23.5 librosa==0.9.2 numba==0.56.4 && \
pip install --no-cache-dir fastapi uvicorn soundfile huggingface_hub ffmpeg-python
# [双重保障-步骤2 / 关键修复]
# 在构建镜像时，预先下载好 NLTK 所需的所有数据包，包括新发现的 "averaged_perceptron_tagger_eng"
RUN python -c "import nltk; nltk.download('punkt', quiet=True, download_dir='/nltk_data'); nltk.download('averaged_perceptron_tagger', quiet=True, download_dir='/nltk_data'); nltk.download('averaged_perceptron_tagger_eng', quiet=True, download_dir='/nltk_data')"
# 预下载依赖模型
COPY download_support_models.py /app/download_support_models.py
RUN python /app/download_support_models.py || true
# --- START OF ADDED SECTION ---
# --- 从 Hugging Face Spaces 下载您自己的模型权重 ---
# 1. 定义权重文件的 URL
ARG GPT_URL="https://huggingface.co/spaces/snsbhg/1111/resolve/main/weights/shantianliang_proplus_e32.ckpt"
ARG SOVITS_URL="https://huggingface.co/spaces/snsbhg/1111/resolve/main/weights/shantianliang_proplus_e8_s192.pth"
# 2. 创建目标目录 (您原来的Dockerfile中已经有COPY指令隐式创建了父目录，这里确保一下)
RUN mkdir -p /app/pretrained_models/shantianliang
# 3. 下载文件到指定目录
RUN echo "--- Downloading model weights ---" && \
wget -nv "$GPT_URL" -O /app/pretrained_models/shantianliang/shantianliang_proplus_e32.ckpt && \
wget -nv "$SOVITS_URL" -O /app/pretrained_models/shantianliang/shantianliang_proplus_e8_s192.pth && \
echo "--- Model weights downloaded successfully ---"
# --- END OF ADDED SECTION ---
# 复制您自己的权重文件和参考音频
# --- 将原来的 COPY weights 指令注释掉或删除 ---
# COPY weights/ /app/pretrained_models/shantianliang/
COPY reference_audio/ /app/reference_audio/
# 更改 /app 目录所有权，赋予运行时？
RUN chown -R 1000:1000 /app
# 暴露 API 端口
EXPOSE 7860
# 容器启动命令
CMD ["python", "api_v2.py", "-a", "0.0.0.0", "-p", "7860", "-c", "GPT_SoVITS/configs/tts_infer.yaml"] |