From d279f5cd1c55a1436b4aa5313e6b69f11730b9f9 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Thu, 16 Jan 2025 17:46:56 +0100
Subject: [PATCH] TRying to check that pb is imported correctly.

---
 Dockerfile | 1 +
 1 file changed, 1 insertion(+)

diff --git a/Dockerfile b/Dockerfile
index a0ec4fe6..d9a00cbe 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -233,6 +233,7 @@ COPY server/Makefile server/Makefile
 ENV UV_SYSTEM_PYTHON=1
 RUN cd server && \
     make gen-server && \
+    python -c "from text_generation_server.pb import generate_pb2" && \
     pip install -U pip uv && \
     uv pip install ".[attention, bnb, accelerate, compressed-tensors, marlin, moe, quantize, peft, outlines]" --no-cache-dir && \
     uv pip install nvidia-nccl-cu12==2.22.3