docs: add offline model download section to README
This commit is contained in:
parent
2e7bbb0cd7
commit
5192355afe
61
README.md
61
README.md
@ -13,6 +13,67 @@ This service provides an async API for separating audio files into vocals and ac
|
|||||||
- **Redis**: Task queue for separation jobs
|
- **Redis**: Task queue for separation jobs
|
||||||
- **Demucs 4.0.1**: AI-powered source separation model (htdemucs)
|
- **Demucs 4.0.1**: AI-powered source separation model (htdemucs)
|
||||||
|
|
||||||
|
## 模型下载(离线部署)
|
||||||
|
|
||||||
|
Demucs 使用 htdemucs 模型(PyTorch hub 格式),首次运行时自动下载,也可手动预下载。
|
||||||
|
|
||||||
|
### 方法1: PyTorch Hub 自动下载(默认)
|
||||||
|
|
||||||
|
服务首次运行时会通过 `torch.hub.load("facebookresearch/demucs", "htdemucs")` 自动下载模型。
|
||||||
|
|
||||||
|
**下载位置**: `~/.cache/torch/hub/checkpoints/`
|
||||||
|
**下载大小**: ~80MB
|
||||||
|
|
||||||
|
### 方法2: 手动预下载
|
||||||
|
|
||||||
|
如果部署环境无法访问外网,可以先在有网络的机器上下载,再拷贝:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 在有网络的机器上运行 Python
|
||||||
|
python3 << "PYTHON"
|
||||||
|
import torch
|
||||||
|
model = torch.hub.load("facebookresearch/demucs", "htdemucs", pretrained=True)
|
||||||
|
print("Model downloaded to:", torch.hub.get_dir())
|
||||||
|
PYTHON
|
||||||
|
|
||||||
|
# 找到模型文件
|
||||||
|
ls ~/.cache/torch/hub/checkpoints/
|
||||||
|
# 应该看到 htdemucs-*.pt 或类似文件
|
||||||
|
|
||||||
|
# 拷贝到部署服务器
|
||||||
|
scp ~/.cache/torch/hub/checkpoints/htdemucs*.pt ymq@opencomputing.net:~/.cache/torch/hub/checkpoints/
|
||||||
|
```
|
||||||
|
|
||||||
|
### 方法3: 直接下载模型文件
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 创建缓存目录
|
||||||
|
mkdir -p ~/.cache/torch/hub/checkpoints/
|
||||||
|
|
||||||
|
# 下载 htdemucs 模型
|
||||||
|
wget -O ~/.cache/torch/hub/checkpoints/htdemucs_v4.pt \
|
||||||
|
https://dl.fbaipublicfiles.com/demucs/v4.0/htdemucs.pth
|
||||||
|
```
|
||||||
|
|
||||||
|
**下载大小**: ~80MB
|
||||||
|
**下载时间**: 约5-15秒
|
||||||
|
|
||||||
|
### 验证下载
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 启动服务后检查日志
|
||||||
|
tail -f /data/ymq/demucs-service/nohup.out | grep -i "model"
|
||||||
|
# 应该看到 "Model loaded" 而不是 "Downloading"
|
||||||
|
```
|
||||||
|
|
||||||
|
### 模型来源
|
||||||
|
|
||||||
|
- **GitHub**: https://github.com/facebookresearch/demucs
|
||||||
|
- **PyTorch Hub**: facebookresearch/demucs
|
||||||
|
- **License**: MIT
|
||||||
|
- **Paper**: Hybrid Spectrogram and Waveform Source Separation
|
||||||
|
|
||||||
|
|
||||||
## API
|
## API
|
||||||
|
|
||||||
### Submit Separation Task
|
### Submit Separation Task
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user