Skip to content
Snippets Groups Projects
Commit a1f69625 authored by Jan Hartig's avatar Jan Hartig
Browse files

Move huggingface cache

parent 4777c680
No related branches found
No related tags found
No related merge requests found
venv
\ No newline at end of file
......@@ -3,9 +3,16 @@ FROM ghcr.io/opennmt/ctranslate2:latest-ubuntu20.04-cuda11.2
# install faster-whisper
RUN python3 -m pip --no-cache-dir install faster-whisper tqdm
# create huggingface home
RUN mkdir /hf
ENV HF_HOME=/hf
# preload model
RUN python3 -c "import faster_whisper; faster_whisper.download_model('large-v2')"
# fix permissions
RUN chmod -R 755 /hf
COPY main.py .
VOLUME "/output/"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment