fix: Correct pip command in Dockerfile for GPU dependencies installation

manascb1344 · Feb 23, 2025 · c69f24b · c69f24b
1 parent 9d06d3d
commit c69f24b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -36,7 +36,7 @@ RUN uv pip install --system -e /app/zonos \
 # Install GPU dependencies with caching
 RUN --mount=type=cache,target=/root/.cache/pip \
     FLASH_ATTENTION_SKIP_CUDA_BUILD=TRUE uv pip install --system flash-attn --no-build-isolation \
-    && uv pip install --system \
+    && 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