Updated wrapt and protobuf versions

Author: jkosek

pyproject.toml

@@ -53,7 +53,6 @@ dependencies = [
     "onnxruntime-gpu~=1.20;platform_machine == 'x86_64' and python_version >= '3.10'",
     "onnxscript",
     "packaging~=24.1",
-    "protobuf>=3.18.3,<3.21.0",
     "polygraphy>=0.49.0",
     "psutil>=5.7",
     "py-cpuinfo>=8.0",
@@ -63,7 +62,7 @@ dependencies = [
     "tabulate>=0.8",
     "tensorrt;platform_machine == 'x86_64'",
     "tritonclient[grpc]~=2.47",
-    "wrapt>=1.11.0,<1.15",
+    "wrapt>=1.16",
     "pyee>=11.1.0",
     "rich>=13.7.1",
 ]
@@ -76,16 +75,19 @@ dependencies = [
 [project.optional-dependencies]
 torch = [
     "torch",
-    "torch_tensorrt>1"
+    "torch_tensorrt>1",
+    "protobuf>=4",
 ]
 tensorflow = [
     "tensorflow",
-    "tf2onnx>=1.9.3"
+    "tf2onnx>=1.9.3",
+    "protobuf>=3.18.3,<3.21.0",
 ]
 jax = [
     "jax[cuda12_pip]",
     "tensorflow",
-    "tf2onnx>=1.9.3"
+    "tf2onnx>=1.9.3",
+    "protobuf>=3.18.3,<3.21.0",
 ]
 test = [
     "GitPython >= 3.1.30",