File size: 382 Bytes
8308b9e bfbe1ae 8308b9e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
# Base image with CUDA support
FROM slabstech/dhwani-model-server:latest
# Create a directory for pre-downloaded models
RUN mkdir -p /app/models
# Define build argument for HF_TOKEN
ARG HF_TOKEN_DOCKER
# Set environment variable for the build process
ENV HF_TOKEN=$HF_TOKEN_DOCKER
# Copy and run the model download script
COPY download_models.py .
RUN python download_models.py |