diff --git a/server/pyproject.toml b/server/pyproject.toml
index 6d0b8dfb..9c2d0183 100644
--- a/server/pyproject.toml
+++ b/server/pyproject.toml
@@ -95,3 +95,6 @@ profile = "black"
 
 [tool.setuptools.packages.find]
 include = ["text_generation_server*"]
+
+[tool.setuptools.package-data]
+"*" = ["pb"]