Spaces:
Running
on
L4
Running
on
L4
Commit
·
7e3ece7
1
Parent(s):
6a8ce38
dock
Browse files- Dockerfile +1 -11
Dockerfile
CHANGED
@@ -24,14 +24,4 @@ RUN PROTOC_ZIP=protoc-21.12-linux-x86_64.zip && \
|
|
24 |
unzip -o $PROTOC_ZIP -d /usr/local 'include/*' && \
|
25 |
rm -f $PROTOC_ZIP
|
26 |
|
27 |
-
|
28 |
-
WORKDIR /home/user
|
29 |
-
RUN git clone https://github.com/JulienDelavande/text-generation-inference.git
|
30 |
-
WORKDIR /home/user/text-generation-inference
|
31 |
-
|
32 |
-
# Installer avec les extensions CUDA
|
33 |
-
ENV BUILD_EXTENSIONS=True
|
34 |
-
RUN make install
|
35 |
-
|
36 |
-
# Lancer le modèle (modifiable dans docker-compose ou commande docker run)
|
37 |
-
CMD ["text-generation-launcher", "--model-id", "mistralai/Mistral-7B-Instruct-v0.2"]
|
|
|
24 |
unzip -o $PROTOC_ZIP -d /usr/local 'include/*' && \
|
25 |
rm -f $PROTOC_ZIP
|
26 |
|
27 |
+
CMD ["tail", "-f", "/dev/null"]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|