diff --git a/pyproject.toml b/pyproject.toml
index bee200b4..1ce5ecd5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,9 +13,10 @@
 #  limitations under the License.
 [project]
 name = "optimum-tpu"
+dynamic = ["version"]
 description = "Optimum TPU is the interface between the Hugging Face Transformers library and Google Cloud TPU devices."
 readme = "README.md"
-license = "LICENSE"
+license = {file = "LICENSE"}
 classifiers = [
     "Development Status :: 1 - Pre-Alpha",
     "License :: OSI Approved :: Apache Software License",
@@ -37,7 +38,12 @@ keywords = [
     "google-cloud"
 ]
 
-dynamic = ["version"]
+dependencies = ["transformers == 4.38.1", ]
+
+[project.optional-dependencies]
+tests = ["pytest", "safetensors"]
+quality = ["black", "ruff", "isort",]
+tpu = ["wheel", "torch-xla>=2.2.0", "torch>=2.2.0",]
 
 [project.urls]
 Homepage = "https://hf.co/hardware"
@@ -46,7 +52,7 @@ Repository = "https://github.com/huggingface/optimum-tpu"
 Issues = "https://github.com/huggingface/optimum-tpu/issues"
 
 [tool.setuptools.dynamic]
-version = {attr = "package.__version__"}
+version = {attr = "optimum.tpu.__version__"}
 
 [tool.black]
 line-length = 119
diff --git a/text-generation-inference/server/pyproject.toml b/text-generation-inference/server/pyproject.toml
index 78d6b0af..30ac683e 100644
--- a/text-generation-inference/server/pyproject.toml
+++ b/text-generation-inference/server/pyproject.toml
@@ -4,14 +4,14 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "text-generation-server"
-version = "VERSION"
+version = "0.1.0.dev0"
 authors = [{name="Alvaro Moran", email="alvaro.moran@huggingface.co" }]
 description = "TGI compatible inference server for Google TPU platforms"
 dependencies = [
-    'protobuf > 3.20.1, < 5',
-    'grpcio == 1.62.0',
-    'grpcio-status == 1.62.0',
-    'grpcio-reflection == 1.62.0',
+    'protobuf',
+    'grpcio == 1.62.1',
+    'grpcio-status == 1.62.1',
+    'grpcio-reflection == 1.62.1',
     'grpc-interceptor == 0.15.2',
     'typer == 0.6.1',
     'safetensors == 0.4.2',