| # 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"] |