Skip to content

vllm.model_executor.models.smolvlm

SmolVLMForConditionalGeneration

Bases: Idefics3ForConditionalGeneration

Source code in vllm/model_executor/models/smolvlm.py 
28
29
30
31
32
33
34
35
36
37
38
@MULTIMODAL_REGISTRY.register_processor(
    SmolVLMMultiModalProcessor,
    info=SmolVLMProcessingInfo,
    dummy_inputs=SmolVLMDummyInputsBuilder,
)
class SmolVLMForConditionalGeneration(Idefics3ForConditionalGeneration):
    def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""):
        super().__init__(
            vllm_config=vllm_config,
            prefix=prefix,
        )

__init__ 

__init__(*, vllm_config: VllmConfig, prefix: str = '')
Source code in vllm/model_executor/models/smolvlm.py 
34
35
36
37
38
def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""):
    super().__init__(
        vllm_config=vllm_config,
        prefix=prefix,
    )

SmolVLMProcessingInfo

Bases: Idefics3ProcessingInfo

Source code in vllm/model_executor/models/smolvlm.py 
15
16
17
18
19
20
21
22
23
24
25
class SmolVLMProcessingInfo(Idefics3ProcessingInfo):
    def get_hf_processor(self, **kwargs: object) -> SmolVLMProcessor:
        return self.ctx.get_hf_processor(SmolVLMProcessor, **kwargs)

    def _get_image_token(self, processor: SmolVLMProcessor | None) -> tuple[str, str]:
        if processor is None:
            processor = self.get_hf_processor()
        image_token = processor.image_token
        fake_image_token = processor.fake_image_token
        global_image_token = processor.global_image_token
        return image_token, fake_image_token, global_image_token

_get_image_token 

_get_image_token(
    processor: SmolVLMProcessor | None,
) -> tuple[str, str]
Source code in vllm/model_executor/models/smolvlm.py 
19
20
21
22
23
24
25
def _get_image_token(self, processor: SmolVLMProcessor | None) -> tuple[str, str]:
    if processor is None:
        processor = self.get_hf_processor()
    image_token = processor.image_token
    fake_image_token = processor.fake_image_token
    global_image_token = processor.global_image_token
    return image_token, fake_image_token, global_image_token

get_hf_processor 

get_hf_processor(**kwargs: object) -> SmolVLMProcessor
Source code in vllm/model_executor/models/smolvlm.py 
16
17
def get_hf_processor(self, **kwargs: object) -> SmolVLMProcessor:
    return self.ctx.get_hf_processor(SmolVLMProcessor, **kwargs)