derek-thomas HF staff commited on
Commit
00d7f53
β€’
1 Parent(s): 5a219d5

Init commit

Browse files
Files changed (3) hide show
  1. .gitignore +1 -0
  2. Dockerfile +28 -0
  3. README.md +1 -1
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ .idea
Dockerfile ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Start from the TGI base image
2
+ FROM [Your TGI Base Image] as base
3
+
4
+ # Install JupyterLab
5
+ # Note: Ensure Python is installed in the base image. If not, you will need to install it.
6
+ RUN pip install jupyterlab
7
+
8
+ # Copy any necessary files (if needed)
9
+ # COPY your-files /your-destination
10
+
11
+ # AWS Sagemaker compatible image
12
+ # Assuming this part remains the same from your original Dockerfile
13
+ FROM base as sagemaker
14
+
15
+ COPY sagemaker-entrypoint.sh entrypoint.sh
16
+ RUN chmod +x entrypoint.sh
17
+
18
+ ENTRYPOINT ["./entrypoint.sh"]
19
+
20
+ # Final image
21
+ FROM base
22
+
23
+ # Add JupyterLab entrypoint
24
+ # Note: You can customize the command to suit your needs
25
+ ENTRYPOINT ["jupyter", "lab", "--ip=0.0.0.0", "--allow-root", "--NotebookApp.token='' --port 7860"]
26
+
27
+ # Optional: Set CMD to launch TGI or any other command
28
+ CMD ["text-generation-launcher", "--json-output"]
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- title: Jais Tgi Benchmark
3
  emoji: πŸ‘
4
  colorFrom: yellow
5
  colorTo: gray
 
1
  ---
2
+ title: Jais TGI Benchmark
3
  emoji: πŸ‘
4
  colorFrom: yellow
5
  colorTo: gray