mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-23 20:45:40 +08:00
23 lines
904 B
Python
23 lines
904 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
# Description: Test the lazy import module
|
|
# The utility function cannot be placed in `vllm.utils`
|
|
# this needs to be a standalone script
|
|
import sys
|
|
|
|
# List of modules that should not be imported too early.
|
|
# Lazy import `torch._inductor.async_compile` to avoid creating
|
|
# too many processes before we set the number of compiler threads.
|
|
# Lazy import `cv2` to avoid bothering users who only use text models.
|
|
# `cv2` can easily mess up the environment.
|
|
module_names = ["torch._inductor.async_compile", "cv2"]
|
|
|
|
# set all modules in `module_names` to be None.
|
|
# if we import any modules during `import vllm`, there would be a
|
|
# hard error and nice stacktrace on the first import.
|
|
for module_name in module_names:
|
|
sys.modules[module_name] = None # type: ignore[assignment]
|
|
|
|
import vllm # noqa
|