fix: finalize pytorch version lock, git lfs

This commit is contained in:
remsky 2024-12-30 06:09:17 -07:00
parent ca66dfe27e
commit 30581129c0

View file

@ -15,11 +15,11 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
# Install heavy Python dependencies first (better layer caching) # Install heavy Python dependencies first (better layer caching)
RUN pip3 install --no-cache-dir \ RUN pip3 install --no-cache-dir \
phonemizer==3.3.0 \ phonemizer==3.3.0 \
torch==2.5.1 \
transformers==4.47.1 \ transformers==4.47.1 \
scipy==1.14.1 \ scipy==1.14.1 \
numpy==2.2.1 \ numpy==2.2.1 \
munch==4.0.0 munch==4.0.0 \
&& pip3 install --no-cache-dir torch==2.5.1 --extra-index-url https://download.pytorch.org/whl/cu121
# Install API dependencies # Install API dependencies
RUN pip3 install --no-cache-dir \ RUN pip3 install --no-cache-dir \
@ -33,15 +33,16 @@ RUN pip3 install --no-cache-dir \
WORKDIR /app WORKDIR /app
# --(can skip if pre-cloning the repo)-- # --(can skip if pre-cloning the repo)--
# Install git-lfs # Install and configure git-lfs
RUN apt-get update && apt-get install -y git-lfs \ RUN apt-get update && apt-get install -y git-lfs \
&& apt-get clean \ && apt-get clean \
&& rm -rf /var/lib/apt/lists/* \ && rm -rf /var/lib/apt/lists/* \
&& git lfs install && git lfs install --skip-repo
# Clone Kokoro repo at specific commit # Clone with LFS
RUN git clone ${KOKORO_REPO} . && \ RUN GIT_LFS_SKIP_SMUDGE=1 git clone ${KOKORO_REPO} . && \
git checkout ${KOKORO_COMMIT} git checkout ${KOKORO_COMMIT} && \
git lfs pull
# -------------------------------------- # --------------------------------------
# Create output directory # Create output directory