Hadeel11 commited on
Commit
33e0dfe
1 Parent(s): e62cb42

Add full fine-tuned model

Browse files
Files changed (1) hide show
  1. Dockerfile.txt +0 -16
Dockerfile.txt DELETED
@@ -1,16 +0,0 @@
1
-
2
- FROM huggingface/transformers-pytorch-gpu:latest
3
-
4
-
5
- RUN pip install --upgrade pip
6
- RUN pip install transformers
7
-
8
-
9
- ENV MODEL_NAME="Hadeel11/your-finetuned-model"
10
- ENV USE_FP16=True
11
-
12
- Expose the port for the inference API
13
- EXPOSE 8080
14
-
15
- Run the TGI server
16
- CMD ["python3", "-m", "transformers_serve", "--model-id", "${MODEL_NAME}"]