Spaces:
Runtime error
Runtime error
Upload 3 files
Browse files- .gitattributes +1 -0
- Dockerfile +50 -0
- download_support_models.py +18 -0
- reference_audio/ref_shantianliang_1.wav +3 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
reference_audio/ref_shantianliang_1.wav filter=lfs diff=lfs merge=lfs -text
|
Dockerfile
ADDED
|
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# 使用官方的、轻量的Python基础镜像
|
| 2 |
+
FROM python:3.9-slim
|
| 3 |
+
|
| 4 |
+
# 在容器内创建一个工作目录
|
| 5 |
+
WORKDIR /code
|
| 6 |
+
|
| 7 |
+
# 安装 wget 用于下载模型
|
| 8 |
+
RUN apt-get update && apt-get install -y --no-install-recommends wget && rm -rf /var/lib/apt/lists/*
|
| 9 |
+
|
| 10 |
+
# 设置环境变量,告诉NLTK将数据存储在/code/nltk_data目录下
|
| 11 |
+
ENV NLTK_DATA /code/nltk_data
|
| 12 |
+
|
| 13 |
+
# 复制依赖文件到工作目录
|
| 14 |
+
COPY ./requirements.txt /code/requirements.txt
|
| 15 |
+
|
| 16 |
+
# 安装依赖。--no-cache-dir 参数可以减小镜像体积
|
| 17 |
+
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
| 18 |
+
|
| 19 |
+
# --- START OF NEW SECTION: 在构建镜像时下载模型 ---
|
| 20 |
+
|
| 21 |
+
# 1. 定义权重文件的 URL
|
| 22 |
+
# 使用 ARG 可以在构建时从外部传入,这里我们直接硬编码
|
| 23 |
+
ARG GPT_URL="https://huggingface.co/spaces/snsbhg/1111/resolve/main/weights/shantianliang_proplus_e32.ckpt"
|
| 24 |
+
ARG SOVITS_URL="https://huggingface.co/spaces/snsbhg/1111/resolve/main/weights/shantianliang_proplus_e8_s192.pth"
|
| 25 |
+
|
| 26 |
+
# 2. 创建目标目录 (与Worker服务中的路径完全一致)
|
| 27 |
+
RUN mkdir -p /app/pretrained_models/shantianliang
|
| 28 |
+
|
| 29 |
+
# 3. 下载文件到指定目录
|
| 30 |
+
# 使用 wget -O 指定输出文件名和路径
|
| 31 |
+
# wget -nv 表示非详细模式,只显示错误
|
| 32 |
+
RUN echo "--- Downloading model weights ---" && \
|
| 33 |
+
wget -nv "$GPT_URL" -O /app/pretrained_models/shantianliang/shantianliang_proplus_e32.ckpt && \
|
| 34 |
+
wget -nv "$SOVITS_URL" -O /app/pretrained_models/shantianliang/shantianliang_proplus_e8_s192.pth && \
|
| 35 |
+
echo "--- Model weights downloaded successfully ---"
|
| 36 |
+
|
| 37 |
+
# --- END OF NEW SECTION ---
|
| 38 |
+
|
| 39 |
+
|
| 40 |
+
# 将应用代码复制到工作目录
|
| 41 |
+
COPY ./app.py /code/app.py
|
| 42 |
+
|
| 43 |
+
# 在构建镜像时,预先下载好 NLTK 数据包
|
| 44 |
+
RUN python -c "import nltk; nltk.download('punkt', quiet=True)"
|
| 45 |
+
|
| 46 |
+
# 暴露端口。Hugging Face Spaces 默认使用 7860 端口
|
| 47 |
+
EXPOSE 7860
|
| 48 |
+
|
| 49 |
+
# 容器启动时要执行的命令
|
| 50 |
+
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
|
download_support_models.py
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from huggingface_hub import snapshot_download
|
| 2 |
+
import os
|
| 3 |
+
|
| 4 |
+
# [关键修改] 将目标文件夹指向程序期望的正确路径
|
| 5 |
+
target = "GPT_SoVITS/pretrained_models"
|
| 6 |
+
os.makedirs(target, exist_ok=True)
|
| 7 |
+
|
| 8 |
+
try:
|
| 9 |
+
snapshot_download(
|
| 10 |
+
repo_id="lj1995/GPT-SoVITS",
|
| 11 |
+
repo_type="model",
|
| 12 |
+
local_dir=target,
|
| 13 |
+
# 注意:这里我们下载全部预训练模型,而不仅仅是sv和chinese
|
| 14 |
+
# allow_patterns=["sv/*", "chinese*"], # 暂时注释掉,确保所有基础模型都被下载
|
| 15 |
+
)
|
| 16 |
+
print("Support models downloaded to ./GPT_SoVITS/pretrained_models")
|
| 17 |
+
except Exception as e:
|
| 18 |
+
print("Skipping support model download:", e)
|
reference_audio/ref_shantianliang_1.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6b373a88f49f2d0d8d9670956d83e155f649d83c8b0ab4b8bcd6ed934b8806c1
|
| 3 |
+
size 395564
|