Update chat.py

chat.py

@@ -159,30 +159,30 @@ class Chat:
             yield json.dumps({"text": generated_text, "error_code": 0}).encode()
 
     def generate_stream_gate(self, params):
-        try:
-            for x in self.generate_stream(params):
-                yield x
-        except ValueError as e:
-            print("Caught ValueError:", e)
-            ret = {
-                "text": server_error_msg,
-                "error_code": 1,
-            }
-            yield json.dumps(ret).encode()
-        except torch.cuda.CudaError as e:
-            print("Caught torch.cuda.CudaError:", e)
-            ret = {
-                "text": server_error_msg,
-                "error_code": 1,
-            }
-            yield json.dumps(ret).encode()
-        except Exception as e:
-            print("Caught Unknown Error", e)
-            ret = {
-                "text": server_error_msg,
-                "error_code": 1,
-            }
-            yield json.dumps(ret).encode()
+        # try:
+        for x in self.generate_stream(params):
+            yield x
+        # except ValueError as e:
+        #     print("Caught ValueError:", e)
+        #     ret = {
+        #         "text": server_error_msg,
+        #         "error_code": 1,
+        #     }
+        #     yield json.dumps(ret).encode()
+        # except torch.cuda.CudaError as e:
+        #     print("Caught torch.cuda.CudaError:", e)
+        #     ret = {
+        #         "text": server_error_msg,
+        #         "error_code": 1,
+        #     }
+        #     yield json.dumps(ret).encode()
+        # except Exception as e:
+        #     print("Caught Unknown Error", e)
+        #     ret = {
+        #         "text": server_error_msg,
+        #         "error_code": 1,
+        #     }
+        #     yield json.dumps(ret).encode()
 
 
 if __name__ == "__main__":