From 14cae49abda9d983c573b2332b4d33a1aa9e1cc7 Mon Sep 17 00:00:00 2001 From: Lucain Pouget Date: Thu, 30 Oct 2025 15:52:35 +0100 Subject: [PATCH 1/2] Close HTTP sessions on fork --- src/huggingface_hub/utils/_http.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/huggingface_hub/utils/_http.py b/src/huggingface_hub/utils/_http.py index 34f80f2811..5b2e2ae8f3 100644 --- a/src/huggingface_hub/utils/_http.py +++ b/src/huggingface_hub/utils/_http.py @@ -17,6 +17,7 @@ import atexit import io import json +import os import re import threading import time @@ -244,6 +245,9 @@ def close_session() -> None: atexit.register(close_session) +if hasattr(os, "register_at_fork"): + os.register_at_fork(after_in_child=close_session) + os.register_at_fork(after_in_parent=close_session) def _http_backoff_base( From e520a9e8a9c0e1b6174b1547a5d8117c68e432d1 Mon Sep 17 00:00:00 2001 From: Lucain Pouget Date: Thu, 30 Oct 2025 15:52:57 +0100 Subject: [PATCH 2/2] only in child fork --- src/huggingface_hub/utils/_http.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/huggingface_hub/utils/_http.py b/src/huggingface_hub/utils/_http.py index 5b2e2ae8f3..75addf5bee 100644 --- a/src/huggingface_hub/utils/_http.py +++ b/src/huggingface_hub/utils/_http.py @@ -247,7 +247,6 @@ def close_session() -> None: atexit.register(close_session) if hasattr(os, "register_at_fork"): os.register_at_fork(after_in_child=close_session) - os.register_at_fork(after_in_parent=close_session) def _http_backoff_base(