biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Misc] Remove import from transformers logging (#5625)

Browse Source
This commit is contained in:
Chang Su
2024-06-18 05:12:19 -07:00
committed by GitHub
parent db5ec52ad7
commit f0cc0e68e3
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
vllm/model_executor/models/phi3v.py
View File

@@ -18,7 +18,6 @@ from typing import Iterable, List, Literal, Optional, Tuple, TypedDict
import torch import torch
import torch.nn as nn import torch.nn as nn
from transformers import CLIPVisionConfig, CLIPVisionModel, PretrainedConfig from transformers import CLIPVisionConfig, CLIPVisionModel, PretrainedConfig
from transformers.utils import logging
from vllm.attention import AttentionMetadata from vllm.attention import AttentionMetadata
from vllm.config import CacheConfig, VisionLanguageConfig from vllm.config import CacheConfig, VisionLanguageConfig
@@ -35,8 +34,6 @@ from vllm.multimodal import MULTIMODAL_REGISTRY
from vllm.multimodal.image import get_dummy_image_data from vllm.multimodal.image import get_dummy_image_data
from vllm.sequence import SamplerOutput from vllm.sequence import SamplerOutput
logger = logging.get_logger(__name__)
_KEYS_TO_MODIFY_MAPPING = { _KEYS_TO_MODIFY_MAPPING = {
"model.vision_embed_tokens": "vision_embed_tokens", "model.vision_embed_tokens": "vision_embed_tokens",
} }