We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent aec61c7 commit 936016cCopy full SHA for 936016c
torchft/checkpointing.py
@@ -16,6 +16,11 @@
16
logger: logging.Logger = logging.getLogger(__name__)
17
18
19
+class IPv6HTTPServer(ThreadingHTTPServer):
20
+ address_family = socket.AF_INET6
21
+ request_queue_size = 1024
22
+
23
24
class CheckpointServer:
25
def __init__(self, state_dict) -> None:
26
self._checkpoint_lock = threading.Lock()
@@ -53,7 +58,7 @@ def err(self, msg: str) -> None:
53
58
self.wfile.write(msg.encode())
54
59
55
60
server_address = ("", 0)
56
- self._server = ThreadingHTTPServer(server_address, RequestHandler)
61
+ self._server = IPv6HTTPServer(server_address, RequestHandler)
57
62
logger.info(f"Started CheckpointServer on {self.address()}...")
63
64
self._thread = threading.Thread(
0 commit comments