heikowagner commited on
Commit
94a2346
1 Parent(s): 65d616f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -19,7 +19,7 @@ RUN python3 convert.py ./models/LLaMA-7B
19
  RUN mkdir ./models/7B/
20
  RUN ./quantize ./models/LLaMA-7B/ggml-model-f16.bin ./models/7B/ggml-model-q4_0.bin q4_0
21
 
22
- FROM tensorflow/tensorflow:latest-gpu
23
  WORKDIR /app
24
  COPY --from=builder /llama.cpp//models/7B/ ./mymodels/LLaMA-7B
25
  # RUN apt-get upgrade -y
 
19
  RUN mkdir ./models/7B/
20
  RUN ./quantize ./models/LLaMA-7B/ggml-model-f16.bin ./models/7B/ggml-model-q4_0.bin q4_0
21
 
22
+ FROM tensorflow/tensorflow:2.11.0-gpu
23
  WORKDIR /app
24
  COPY --from=builder /llama.cpp//models/7B/ ./mymodels/LLaMA-7B
25
  # RUN apt-get upgrade -y