Skip to content

vllm.model_executor.model_loader.dummy_loader

DummyModelLoader

Bases: BaseModelLoader

Model loader that will set model weights to random values.

Source code in vllm/model_executor/model_loader/dummy_loader.py 
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
class DummyModelLoader(BaseModelLoader):
    """Model loader that will set model weights to random values."""

    def __init__(self, load_config: LoadConfig):
        super().__init__(load_config)
        if load_config.model_loader_extra_config:
            raise ValueError(
                f"Model loader extra config is not supported for "
                f"load format {load_config.load_format}"
            )

    def download_model(self, model_config: ModelConfig) -> None:
        pass  # Nothing to download

    def load_weights(self, model: nn.Module, model_config: ModelConfig) -> None:
        # NOTE(woosuk): For accurate performance evaluation, we assign
        # random values to the weights.
        initialize_dummy_weights(model)

__init__ 

__init__(load_config: LoadConfig)
Source code in vllm/model_executor/model_loader/dummy_loader.py 
14
15
16
17
18
19
20
def __init__(self, load_config: LoadConfig):
    super().__init__(load_config)
    if load_config.model_loader_extra_config:
        raise ValueError(
            f"Model loader extra config is not supported for "
            f"load format {load_config.load_format}"
        )

download_model 

download_model(model_config: ModelConfig) -> None
Source code in vllm/model_executor/model_loader/dummy_loader.py 
22
23
def download_model(self, model_config: ModelConfig) -> None:
    pass  # Nothing to download

load_weights 

load_weights(
    model: Module, model_config: ModelConfig
) -> None
Source code in vllm/model_executor/model_loader/dummy_loader.py 
25
26
27
28
def load_weights(self, model: nn.Module, model_config: ModelConfig) -> None:
    # NOTE(woosuk): For accurate performance evaluation, we assign
    # random values to the weights.
    initialize_dummy_weights(model)