Update model_service_worker.py (#2809)

pytorch · Nov 23, 2023 · f5d868e · f5d868e
1 parent 2e1d7c6
commit f5d868e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ts/model_service_worker.py b/ts/model_service_worker.py
@@ -177,6 +177,7 @@ def handle_connection(self, cl_socket):
             cmd, msg = retrieve_msg(cl_socket)
             if BENCHMARK:
                 pr.enable()
+            # b"I" encodes a model Inference request from frontend
             if cmd == b"I":
                 if service is not None:
                     resp = service.predict(msg)
@@ -185,6 +186,7 @@ def handle_connection(self, cl_socket):
                     raise RuntimeError(
                         "Received command: {}, but service is not loaded".format(cmd)
                     )
+            # b"I" encodes a model Loading request from frontend
             elif cmd == b"L":
                 service, result, code = self.load_model(msg)
                 resp = bytearray()