From 39583ec639364bc373f1db1c4ecf74e80a8eab76 Mon Sep 17 00:00:00 2001 From: "Xingjun.Wang" Date: Thu, 25 Jun 2026 16:04:11 +0800 Subject: [PATCH] fix: patch_context() use try/finally to ensure unpatch on exception (#1743) - Wrapped yield in try/finally block so unpatch_hub() executes even when the with-body raises an exception, preventing patch state leakage --- modelscope/utils/hf_util/patcher.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modelscope/utils/hf_util/patcher.py b/modelscope/utils/hf_util/patcher.py index e58e84b6..e4266e08 100644 --- a/modelscope/utils/hf_util/patcher.py +++ b/modelscope/utils/hf_util/patcher.py @@ -1014,5 +1014,7 @@ def unpatch_hub(): @contextlib.contextmanager def patch_context(): patch_hub() - yield - unpatch_hub() + try: + yield + finally: + unpatch_hub()