Skip to content

Commit c261a02

Browse files
committed
update model pre-download
1 parent 09a10e9 commit c261a02

3 files changed

Lines changed: 8 additions & 1 deletion

File tree

Dockerfile

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,3 +11,5 @@ RUN pip install --no-cache-dir \
1111

1212
COPY . .
1313

14+
RUN python download_model.py
15+

docker-compose.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ services:
44
build:
55
context: .
66
dockerfile: Dockerfile
7-
image: paidax/funasr_python:0.3.1
7+
image: paidax/funasr_python:0.3.2
88
runtime: nvidia
99
ports:
1010
- 9527:9527

download_model.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
from modelscope.hub.snapshot_download import snapshot_download
2+
3+
snapshot_download("damo/speech_fsmn_vad_zh-cn-16k-common-pytorch")
4+
snapshot_download("damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch")
5+
snapshot_download("damo/punc_ct-transformer_zh-cn-common-vocab272727-pytorch")

0 commit comments

Comments
 (0)