SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Better error message when signal is not supported on OS (#25049)

Browse Source 
* Better error message when signal is not supported on OS

* Address review comments
This commit is contained in:
Sylvain Gugger
2023-07-24 14:34:16 -04:00
committed by GitHub
parent c0d1c33022
commit d2295708a6
1 changed files with 20 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/transformers/dynamic_module_utils.py
View File

@@ -532,6 +532,7 @@ def resolve_trust_remote_code(trust_remote_code, model_name, has_local_code, has
if has_local_code: if has_local_code:
trust_remote_code = False trust_remote_code = False
elif has_remote_code and TIME_OUT_REMOTE_CODE > 0: elif has_remote_code and TIME_OUT_REMOTE_CODE > 0:
try:
signal.signal(signal.SIGALRM, _raise_timeout_error) signal.signal(signal.SIGALRM, _raise_timeout_error)
signal.alarm(TIME_OUT_REMOTE_CODE) signal.alarm(TIME_OUT_REMOTE_CODE)
while trust_remote_code is None: while trust_remote_code is None:
@@ -545,6 +546,13 @@ def resolve_trust_remote_code(trust_remote_code, model_name, has_local_code, has
elif answer.lower() in ["no", "n", "0", ""]: elif answer.lower() in ["no", "n", "0", ""]:
trust_remote_code = False trust_remote_code = False
signal.alarm(0) signal.alarm(0)
except AttributeError:
# 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. "
f"Make sure you have read the code at https://hf.co/{model_name} to avoid malicious use, then set "
"the option `trust_remote_code=True` to remove this error."
)
elif has_remote_code: elif has_remote_code:
# For the CI which puts the timeout at 0 # For the CI which puts the timeout at 0
_raise_timeout_error(None, None) _raise_timeout_error(None, None)