mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-28 09:08:54 +00:00
add seed-vc model (#60)
* init seed-vc * seed-vc workflow * put seed-vc in the right dir
This commit is contained in:
+4
-11
@@ -1,5 +1,5 @@
|
||||
# Base CUDA image
|
||||
FROM cnstark/pytorch:2.3.1-py3.10.15-cuda12.1.0-devel-ubuntu22.04
|
||||
FROM pytorch/pytorch:2.5.0-cuda12.1-cudnn9-runtime
|
||||
|
||||
# Set environment variables
|
||||
ENV DEBIAN_FRONTEND=noninteractive
|
||||
@@ -14,13 +14,7 @@ RUN apt-get update && \
|
||||
git \
|
||||
nano \
|
||||
curl \
|
||||
software-properties-common \
|
||||
sudo \
|
||||
rsync && \
|
||||
curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | bash && \
|
||||
apt-get install -y git-lfs && \
|
||||
git lfs install && \
|
||||
add-apt-repository ppa:deadsnakes/ppa && \
|
||||
software-properties-common && \
|
||||
rm -rf /var/lib/apt/lists/*
|
||||
|
||||
# Create directories for Python installation and model code
|
||||
@@ -43,14 +37,13 @@ COPY . /model_code/F5-TTS
|
||||
|
||||
# /model_code/F5-TTS/pretrained_models
|
||||
|
||||
|
||||
# initializes the model - downloads checkpoints
|
||||
# FIXME: is isn't working
|
||||
ARG HF_DATASETS_CACHE="./pretrained_models"
|
||||
# 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"]
|
||||
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"]
|
||||
|
||||
Reference in New Issue
Block a user