biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Fix offline test for Transformers v5 (#33682)

Browse Source 
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
Harry Mellor
2026-02-03 12:07:24 +00:00
committed by GitHub
parent ceab70c89d
commit f6af34626d
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/entrypoints/offline_mode/test_offline_mode.py
View File

@@ -109,8 +109,15 @@ def _re_import_modules():
if k.startswith("transformers") and not k.startswith("transformers_modules") if k.startswith("transformers") and not k.startswith("transformers_modules")
] ]
# These modules are aliased in Transformers v5 and so cannot be reloaded directly
aliased_modules = ["tokenization_utils", "tokenization_utils_fast"]
reload_exception = None reload_exception = None
for module_name in hf_hub_module_names + transformers_module_names: for module_name in hf_hub_module_names + transformers_module_names:
if any(module_name.endswith(f".{alias}") for alias in aliased_modules):
# Remove from sys.modules so they are re-aliased on next import
del sys.modules[module_name]
continue
try: try:
importlib.reload(sys.modules[module_name]) importlib.reload(sys.modules[module_name])
except Exception as e: except Exception as e: