teoha commited on
Commit
2bd44dd
·
1 Parent(s): e7203d2

Migrate to use small-q5_1 for faster inference

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -6,7 +6,7 @@ RUN mkdir /whisper && \
6
 
7
  WORKDIR /whisper
8
 
9
- ARG model=medium-q5_0
10
  RUN bash ./models/download-ggml-model.sh "${model}"
11
  RUN make main
12
 
@@ -26,7 +26,7 @@ RUN pip install -r $HOME/root/requirements.txt
26
  COPY --chown=user main.py $HOME/root
27
  COPY --chown=user youtubeaudio.py $HOME/root
28
 
29
- ARG model=medium-q5_0
30
  ENV model=$model
31
  RUN mkdir $HOME/root/models
32
  RUN mkdir -p -m 775 /tmp/holosubs/tracks /tmp/holosubs/videos
 
6
 
7
  WORKDIR /whisper
8
 
9
+ ARG model=small-q5_1
10
  RUN bash ./models/download-ggml-model.sh "${model}"
11
  RUN make main
12
 
 
26
  COPY --chown=user main.py $HOME/root
27
  COPY --chown=user youtubeaudio.py $HOME/root
28
 
29
+ ARG model=small-q5_1
30
  ENV model=$model
31
  RUN mkdir $HOME/root/models
32
  RUN mkdir -p -m 775 /tmp/holosubs/tracks /tmp/holosubs/videos