Skip to content

GitLab

Commit 22c4fd07 authored by OlivierDehaene's avatar OlivierDehaene
Browse files

fix(docker): use ubuntu20.04

parent 119f7e06
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 3 deletions
Dockerfile
View file @ 22c4fd07
...@@ -108,7 +108,7 @@ COPY server/Makefile-transformers Makefile ...@@ -108,7 +108,7 @@ COPY server/Makefile-transformers Makefile
RUN BUILD_EXTENSIONS="True" make build-transformers RUN BUILD_EXTENSIONS="True" make build-transformers
# Text Generation Inference base image # Text Generation Inference base image
FROM nvidia/cuda:11.8.0-base-ubuntu22.04 as base FROM nvidia/cuda:11.8.0-base-ubuntu20.04 as base
# Conda env # Conda env
ENV PATH=/opt/conda/bin:$PATH \ ENV PATH=/opt/conda/bin:$PATH \
...@@ -117,8 +117,6 @@ ENV PATH=/opt/conda/bin:$PATH \ ...@@ -117,8 +117,6 @@ ENV PATH=/opt/conda/bin:$PATH \
# Text Generation Inference base env # Text Generation Inference base env
ENV HUGGINGFACE_HUB_CACHE=/data \ ENV HUGGINGFACE_HUB_CACHE=/data \
HF_HUB_ENABLE_HF_TRANSFER=1 \ HF_HUB_ENABLE_HF_TRANSFER=1 \
MODEL_ID=bigscience/bloom-560m \
NUM_SHARD=1 \
PORT=80 PORT=80
WORKDIR /usr/src WORKDIR /usr/src
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment