From f92cc7034a49959b247a46a210b912e56a6f977d Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Mon, 21 Aug 2023 09:01:11 +0200 Subject: [PATCH] Ignore all exceptions from signal in dynamic code (#25623) --- src/transformers/dynamic_module_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/dynamic_module_utils.py b/src/transformers/dynamic_module_utils.py index 031ab8b7d1..611702baeb 100644 --- a/src/transformers/dynamic_module_utils.py +++ b/src/transformers/dynamic_module_utils.py @@ -603,7 +603,7 @@ def resolve_trust_remote_code(trust_remote_code, model_name, has_local_code, has elif answer.lower() in ["no", "n", "0", ""]: trust_remote_code = False signal.alarm(0) - except AttributeError: + except Exception: # OS which does not support signal.SIGALRM raise ValueError( "Loading this model requires you to execute execute some code in that repo on your local machine. "