File size: 828 Bytes
00d7f53 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
# Start from the TGI base image
FROM [Your TGI Base Image] as base
# Install JupyterLab
# Note: Ensure Python is installed in the base image. If not, you will need to install it.
RUN pip install jupyterlab
# Copy any necessary files (if needed)
# COPY your-files /your-destination
# AWS Sagemaker compatible image
# Assuming this part remains the same from your original Dockerfile
FROM base as sagemaker
COPY sagemaker-entrypoint.sh entrypoint.sh
RUN chmod +x entrypoint.sh
ENTRYPOINT ["./entrypoint.sh"]
# Final image
FROM base
# Add JupyterLab entrypoint
# Note: You can customize the command to suit your needs
ENTRYPOINT ["jupyter", "lab", "--ip=0.0.0.0", "--allow-root", "--NotebookApp.token='' --port 7860"]
# Optional: Set CMD to launch TGI or any other command
CMD ["text-generation-launcher", "--json-output"]
|