Increase TFS payload max length to 128 MB

cortexlabs · Dec 25, 2020 · 652924b · 652924b
1 parent 888df08
commit 652924b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkg/cortex/serve/cortex_internal/lib/model/tfs.py b/pkg/cortex/serve/cortex_internal/lib/model/tfs.py
@@ -113,7 +113,8 @@ def __init__(self, address: str):
             {}
         )  # maps the model ID to the model metadata (signature def, signature key and so on)
 
-        self.channel = grpc.insecure_channel(self.address)
+        options = [('grpc.max_message_length', 128 * 1024 * 1024)]
+        self.channel = grpc.insecure_channel(self.address, options=options)
         self._service = model_service_pb2_grpc.ModelServiceStub(self.channel)
         self._pred = prediction_service_pb2_grpc.PredictionServiceStub(self.channel)