Unverified Commit ba2910f7 authored by Harry Mellor's avatar Harry Mellor Committed by GitHub
Browse files

Fix offline mode test for Transformers v5 (#38095)


Signed-off-by: default avatarHarry Mellor <19981378+hmellor@users.noreply.github.com>
parent f262a62a
......@@ -7,6 +7,7 @@ import importlib
import sys
import pytest
import regex as re
import urllib3
from vllm import LLM
......@@ -110,11 +111,15 @@ def _re_import_modules():
]
# These modules are aliased in Transformers v5 and so cannot be reloaded directly
aliased_modules = ["tokenization_utils", "tokenization_utils_fast"]
aliased_module_patterns = [
r".+\.tokenization_utils$",
r".+\.tokenization_utils_fast$",
r".+\.models\..+\.image_processing_.+_fast$",
]
reload_exception = None
for module_name in hf_hub_module_names + transformers_module_names:
if any(module_name.endswith(f".{alias}") for alias in aliased_modules):
if any(re.match(pattern, module_name) for pattern in aliased_module_patterns):
# Remove from sys.modules so they are re-aliased on next import
del sys.modules[module_name]
continue
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment