mhussainahmad
/

sdxl-wombo-optimized

Model card Files Files and versions Community

Ashley Wright commited on Sep 8, 2024

Commit

921dd6f

•

1 Parent(s): fcd4555

Handle EOF

Files changed (1) hide show

src/main.py +6 -1

src/main.py CHANGED Viewed

@@ -27,7 +27,12 @@ def main():
             print(f"Connected")
             while True:
-                request = TextToImageRequest.model_validate_json(connection.recv_bytes().decode("utf-8"))
                 image = infer(request, pipeline)

             print(f"Connected")
             while True:
+                try:
+                    request = TextToImageRequest.model_validate_json(connection.recv_bytes().decode("utf-8"))
+                except EOFError:
+                    print(f"Inference socket exiting")
+                    return
                 image = infer(request, pipeline)