storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-17 00:23:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

f5-tts: remove HF_DATASET_CACHE from docker build

Browse Source
This commit is contained in:
Madhukar Mishra
2024-10-21 18:59:46 +05:30
parent b9c2c57690
commit 9fd8912e95
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/F5-TTS/Dockerfile
+7 -4
View File
@@ -42,12 +42,15 @@ WORKDIR /model_code/F5-TTS
COPY . /model_code/F5-TTS
# /model_code/F5-TTS/pretrained_models
ARG HF_DATASETS_CACHE="./pretrained_models"
# FIXME: this might not work without HF_TOKEN!!!
ENV HF_DATASETS_CACHE=$HF_DATASETS_CACHE
# initializes the model - downloads checkpoints
RUN ["/bin/bash", "-c", "HF_DATASETS_CACHE=$HF_DATASETS_CACHE DOWNLOAD_ONLY=TRUE /python_install/python/bin/python inference-cli.py"]
# FIXME: is isn't working
ARG HF_DATASETS_CACHE="./pretrained_models"
# ENV HF_DATASETS_CACHE=$HF_DATASETS_CACHE
# RUN ["/bin/bash", "-c", "HF_DATASETS_CACHE=$HF_DATASETS_CACHE DOWNLOAD_ONLY=TRUE /python_install/python/bin/python inference-cli.py"]
# checkpoints at the default location ~/.cache/huggingface/hub
RUN ["/bin/bash", "-c", "DOWNLOAD_ONLY=TRUE /python_install/python/bin/python inference-cli.py"]
ENTRYPOINT ["/bin/bash", "-c", ". /python_install/python/bin/activate && exec /bin/bash"]