Convert formatting to use ruff instead of yapf + isort (#26247)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
@@ -10,20 +10,19 @@ from vllm.multimodal import MULTIMODAL_REGISTRY
|
||||
|
||||
# yapf: disable
|
||||
from .idefics3 import Idefics3DummyInputsBuilder as SmolVLMDummyInputsBuilder
|
||||
from .idefics3 import Idefics3ForConditionalGeneration
|
||||
from .idefics3 import Idefics3ForConditionalGeneration, Idefics3ProcessingInfo
|
||||
from .idefics3 import Idefics3MultiModalProcessor as SmolVLMMultiModalProcessor
|
||||
from .idefics3 import Idefics3ProcessingInfo
|
||||
|
||||
# yapf: enable
|
||||
|
||||
|
||||
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: Optional[SmolVLMProcessor]) -> tuple[str, str]:
|
||||
self, processor: Optional[SmolVLMProcessor]
|
||||
) -> tuple[str, str]:
|
||||
if processor is None:
|
||||
processor = self.get_hf_processor()
|
||||
image_token = processor.image_token
|
||||
@@ -32,11 +31,12 @@ class SmolVLMProcessingInfo(Idefics3ProcessingInfo):
|
||||
return image_token, fake_image_token, global_image_token
|
||||
|
||||
|
||||
@MULTIMODAL_REGISTRY.register_processor(SmolVLMMultiModalProcessor,
|
||||
info=SmolVLMProcessingInfo,
|
||||
dummy_inputs=SmolVLMDummyInputsBuilder)
|
||||
@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,
|
||||
|
||||
Reference in New Issue
Block a user