Skip to content

Commit

Permalink
feat: Update Dockerfile to use --system flag for pip installations an…
Browse files Browse the repository at this point in the history
…d improve dependency management
  • Loading branch information
manascb1344 committed Feb 23, 2025
1 parent 5ef7e78 commit 9d06d3d
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
espeak-ng \
&& rm -rf /var/lib/apt/lists/* \
&& pip install uv \
&& uv pip install -r requirements.txt
&& uv pip install --system -r requirements.txt

# Clone the zonos repository and update submodules
RUN git clone https://github.com/Zyphra/Zonos.git /app/zonos \
Expand All @@ -27,16 +27,16 @@ RUN git clone https://github.com/Zyphra/Zonos.git /app/zonos \

# Install Zonos and its dependencies
RUN uv pip install --system -e /app/zonos \
&& uv pip install kanjize>=1.5.0 \
&& uv pip install --system kanjize>=1.5.0 \
inflect>=7.5.0 \
phonemizer>=3.3.0 \
sudachidict-full>=20241021 \
sudachipy>=0.6.10

# Install GPU dependencies with caching
RUN --mount=type=cache,target=/root/.cache/pip \
FLASH_ATTENTION_SKIP_CUDA_BUILD=TRUE uv pip install flash-attn --no-build-isolation \
&& uv pip install --no-cache-dir \
FLASH_ATTENTION_SKIP_CUDA_BUILD=TRUE uv pip install --system flash-attn --no-build-isolation \
&& uv pip install --system \
https://github.com/state-spaces/mamba/releases/download/v2.2.4/mamba_ssm-2.2.4+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl \
https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.5.0.post8/causal_conv1d-1.5.0.post8+cu12torch2.4cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

Expand Down

0 comments on commit 9d06d3d

Please sign in to comment.