teoha commited on
Commit
595b296
·
1 Parent(s): 4fde126

Pick small-q5_1 after testing different models

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=base-q5_1
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=base-q5_1
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