Tidy vllm/config/__init__.py to only add classes and functions (#26405)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
@@ -1,42 +1,28 @@
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
||||
|
||||
from vllm.config.cache import (
|
||||
BlockSize,
|
||||
CacheConfig,
|
||||
CacheDType,
|
||||
MambaDType,
|
||||
PrefixCachingHashAlgo,
|
||||
)
|
||||
from vllm.config.cache import CacheConfig
|
||||
from vllm.config.compilation import (
|
||||
CompilationConfig,
|
||||
CompilationLevel,
|
||||
CUDAGraphMode,
|
||||
PassConfig,
|
||||
)
|
||||
from vllm.config.device import Device, DeviceConfig
|
||||
from vllm.config.device import DeviceConfig
|
||||
from vllm.config.kv_events import KVEventsConfig
|
||||
from vllm.config.kv_transfer import KVTransferConfig
|
||||
from vllm.config.load import LoadConfig
|
||||
from vllm.config.lora import LoRAConfig
|
||||
from vllm.config.model import (
|
||||
ConvertOption,
|
||||
HfOverrides,
|
||||
LogprobsMode,
|
||||
ModelConfig,
|
||||
ModelDType,
|
||||
ModelImpl,
|
||||
RunnerOption,
|
||||
TaskOption,
|
||||
TokenizerMode,
|
||||
iter_architecture_defaults,
|
||||
try_match_architecture_defaults,
|
||||
)
|
||||
from vllm.config.multimodal import MMCacheType, MMEncoderTPMode, MultiModalConfig
|
||||
from vllm.config.observability import DetailedTraceModules, ObservabilityConfig
|
||||
from vllm.config.parallel import DistributedExecutorBackend, EPLBConfig, ParallelConfig
|
||||
from vllm.config.multimodal import MultiModalConfig
|
||||
from vllm.config.observability import ObservabilityConfig
|
||||
from vllm.config.parallel import EPLBConfig, ParallelConfig
|
||||
from vllm.config.pooler import PoolerConfig
|
||||
from vllm.config.scheduler import RunnerType, SchedulerConfig, SchedulerPolicy
|
||||
from vllm.config.scheduler import SchedulerConfig
|
||||
from vllm.config.speculative import SpeculativeConfig
|
||||
from vllm.config.speech_to_text import SpeechToTextConfig
|
||||
from vllm.config.structured_outputs import StructuredOutputsConfig
|
||||
@@ -56,20 +42,17 @@ from vllm.config.vllm import (
|
||||
set_current_vllm_config,
|
||||
)
|
||||
|
||||
# __all__ should only contain classes and functions.
|
||||
# Types and globals should be imported from their respective modules.
|
||||
__all__ = [
|
||||
# From vllm.config.cache
|
||||
"BlockSize",
|
||||
"CacheConfig",
|
||||
"CacheDType",
|
||||
"MambaDType",
|
||||
"PrefixCachingHashAlgo",
|
||||
# From vllm.config.compilation
|
||||
"CompilationConfig",
|
||||
"CompilationLevel",
|
||||
"CUDAGraphMode",
|
||||
"PassConfig",
|
||||
# From vllm.config.device
|
||||
"Device",
|
||||
"DeviceConfig",
|
||||
# From vllm.config.kv_events
|
||||
"KVEventsConfig",
|
||||
@@ -80,34 +63,20 @@ __all__ = [
|
||||
# From vllm.config.lora
|
||||
"LoRAConfig",
|
||||
# From vllm.config.model
|
||||
"ConvertOption",
|
||||
"HfOverrides",
|
||||
"LogprobsMode",
|
||||
"ModelConfig",
|
||||
"ModelDType",
|
||||
"ModelImpl",
|
||||
"RunnerOption",
|
||||
"TaskOption",
|
||||
"TokenizerMode",
|
||||
"iter_architecture_defaults",
|
||||
"try_match_architecture_defaults",
|
||||
# From vllm.config.multimodal
|
||||
"MMCacheType",
|
||||
"MMEncoderTPMode",
|
||||
"MultiModalConfig",
|
||||
# From vllm.config.observability
|
||||
"DetailedTraceModules",
|
||||
"ObservabilityConfig",
|
||||
# From vllm.config.parallel
|
||||
"DistributedExecutorBackend",
|
||||
"EPLBConfig",
|
||||
"ParallelConfig",
|
||||
# From vllm.config.pooler
|
||||
"PoolerConfig",
|
||||
# From vllm.config.scheduler
|
||||
"RunnerType",
|
||||
"SchedulerConfig",
|
||||
"SchedulerPolicy",
|
||||
# From vllm.config.speculative
|
||||
"SpeculativeConfig",
|
||||
# From vllm.config.speech_to_text
|
||||
|
||||
Reference in New Issue
Block a user