Skip to content

Commit aba7fbf

Browse files
Starting (#2)
* Windows is a pain * Windows is a pain 2 * Hmm at least its working now. Time to add some UI * Some working UI with warmup .. i need to change that and make it better. * A bit better. More work needed. Time to eat. * A bit better. More work needed. * Again progress. Stable version. * Further progress. Now fonts. * Further progress * UI is good enough lets focus on api. * Working api and UI and added queue for tasks. * Small cleanup before next round. * Few upgrades to the UI and API. Working on defaults. * Corrected default issue and locked dependencies. * Working initial build. * Fix the current build while working on whl
1 parent 9ff603c commit aba7fbf

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

Dockerfile

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
# Stage 1: Build environment with dependencies and models
33
# ============================================================
44
FROM python:3.10-slim AS builder
5+
#FROM pytorch/pytorch:2.8.0-cuda12.9-cudnn9-devel AS builder
56

67
WORKDIR /app
78

@@ -19,8 +20,12 @@ RUN pip install --upgrade pip setuptools wheel
1920
RUN pip install -r requirements.txt
2021
# RUN python -m unidic download
2122

23+
# Install matching prebuilt wheel
24+
#RUN pip install https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.3.17/flash_attn-2.7.4+cu129torch2.8-cp311-cp311-linux_x86_64.whl
25+
26+
2227
# Install Flash Attention for faster attention mechanisms
23-
RUN pip install flash-attn --no-build-isolation
28+
# RUN pip install flash-attn --no-build-isolation
2429

2530
# Now bring in the actual application code
2631
COPY app/ .
@@ -38,6 +43,7 @@ RUN python init_downloads.py && \
3843
# Stage 2: Final runtime image
3944
# ============================================================
4045
FROM python:3.10-slim
46+
#FROM pytorch/pytorch:2.8.0-cuda12.9-cudnn9-devel
4147

4248
WORKDIR /app
4349

@@ -46,6 +52,7 @@ RUN apt-get update && apt-get install -y \
4652
&& rm -rf /var/lib/apt/lists/*
4753

4854
COPY --from=builder /usr/local /usr/local
55+
#COPY --from=builder /opt/conda /opt/conda
4956
COPY --from=builder /app /app
5057
COPY --from=builder /root/.cache/huggingface /root/.cache/huggingface
5158
# COPY --from=builder /root/nltk_data /root/nltk_data

0 commit comments

Comments
 (0)