Skip to content

Commit

Permalink
Make internal
Browse files Browse the repository at this point in the history
Signed-off-by: DarkLight1337 <[email protected]>
  • Loading branch information
DarkLight1337 committed Dec 30, 2024
1 parent ed88a2d commit e7aa0f1
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions vllm/multimodal/parse.py
Original file line number Diff line number Diff line change
Expand Up @@ -227,7 +227,7 @@ def __init__(self, *, target_sr: Optional[float] = None) -> None:

self.target_sr = target_sr

def is_embeddings(self, data: object) -> TypeGuard[NestedTensors]:
def _is_embeddings(self, data: object) -> TypeGuard[NestedTensors]:
if isinstance(data, torch.Tensor):
return data.ndim == 3
if is_list_of(data, torch.Tensor):
Expand All @@ -254,7 +254,7 @@ def _parse_audio_data(
self,
data: ModalityData[AudioItem],
) -> ModalityDataItems[Any, Any]:
if self.is_embeddings(data):
if self._is_embeddings(data):
return AudioEmbeddingsInput(data)

if (is_list_of(data, float)
Expand Down Expand Up @@ -291,7 +291,7 @@ def _parse_image_data(
self,
data: ModalityData[ImageItem],
) -> ModalityDataItems[Any, Any]:
if self.is_embeddings(data):
if self._is_embeddings(data):
return ImageEmbeddingsInput(data)

if (isinstance(data, Image)
Expand All @@ -309,7 +309,7 @@ def _parse_video_data(
self,
data: ModalityData[VideoItem],
) -> ModalityDataItems[Any, Any]:
if self.is_embeddings(data):
if self._is_embeddings(data):
return VideoEmbeddingsInput(data)

if (is_list_of(data, Image)
Expand Down

0 comments on commit e7aa0f1

Please sign in to comment.