File size: 407 Bytes
bb38358 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
# Start from a PyTorch + CUDA base image
FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-devel
WORKDIR /code
COPY . .
# Install dependencies
RUN pip install -r requirements.txt
RUN pip install flash-attn --no-build-isolation
# Make the startup script executable
COPY startup.sh .
RUN chmod +x startup.sh
# This is the new command: it runs your script automatically when the Space starts.
CMD ["./startup.sh"] |