[Model] Support DeepSeek-V4

This commit is contained in:
chenxb002
2026-04-24 09:50:34 +08:00
commit b9925203b8
172 changed files with 44780 additions and 0 deletions

View File

@@ -0,0 +1,3 @@
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-MLU project

View File

@@ -0,0 +1,46 @@
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-MLU project
from typing import Any, TypeVar
from PIL import Image
from vllm import multimodal
from vllm.logger import init_logger
from vllm.multimodal.utils import MediaConnector
from vllm_mlu.mlu_hijack_utils import MluHijackObject
def vllm__multimodal__utils__fetch_image(
image_url: str,
image_io_kwargs: dict[str, Any] | None = None,
) -> Image.Image:
"""
Args:
image_url: URL of the image file to fetch.
image_io_kwargs: Additional kwargs passed to handle image IO.
"""
media_io_kwargs = None if not image_io_kwargs else {"image": image_io_kwargs}
'''
=============================
Modify by vllm_mlu
=============================
@brief: set 'allowed_local_media_path' as default
'''
media_connector = MediaConnector(media_io_kwargs,
allowed_local_media_path=image_io_kwargs["allowed_local_media_path"]
)
'''
==================
End of MLU Hijack
==================
'''
return media_connector.fetch_image(image_url)
MluHijackObject.apply_hijack(multimodal,
multimodal.utils,
vllm__multimodal__utils__fetch_image)