wan2gp / shared /asyncio_utils.py
vidfom's picture
Upload folder using huggingface_hub
618f472 verified
import os
def silence_proactor_connection_reset() -> None:
if os.name != "nt":
return
from asyncio import proactor_events as _proactor
transport_cls = _proactor._ProactorBasePipeTransport
if getattr(transport_cls, "_wan2gp_patch", False):
return
original = transport_cls._call_connection_lost
def _call_connection_lost(self, exc):
if isinstance(exc, ConnectionResetError):
exc = None
try:
return original(self, exc)
except ConnectionResetError:
return None
transport_cls._call_connection_lost = _call_connection_lost
transport_cls._wan2gp_patch = True