Handle broken pipe error (#572)

Author: mahou-shoujo

koboldcpp.py

@@ -626,7 +626,7 @@ async def handle_request(self, genparams, api_format, stream_flag):
             await asyncio.gather(*tasks)
             generate_result = generate_task.result()
             return generate_result
-        except ConnectionAbortedError as cae: # attempt to abort if connection lost
+        except (BrokenPipeError, ConnectionAbortedError) as cae: # attempt to abort if connection lost
             print(cae)
             handle.abort_generate()
             time.sleep(0.1) #short delay