Skip to content

GitLab

Unverified Commit 06386a64 authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

[Frontend] Chat-based Embeddings API (#9759)

parent d3aa2a8b
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
vllm/pooling_params.py
View file @ 06386a64
...@@ -7,7 +7,7 @@ class PoolingParams( ...@@ -7,7 +7,7 @@ class PoolingParams(
msgspec.Struct, msgspec.Struct,
omit_defaults=True, # type: ignore[call-arg] omit_defaults=True, # type: ignore[call-arg]
array_like=True): # type: ignore[call-arg] array_like=True): # type: ignore[call-arg]
"""Pooling parameters for pooling. """Pooling parameters for embeddings API.
Attributes: Attributes:
additional_data: Any additional data needed for pooling. additional_data: Any additional data needed for pooling.
...@@ -16,7 +16,7 @@ class PoolingParams( ...@@ -16,7 +16,7 @@ class PoolingParams(
def clone(self) -> "PoolingParams": def clone(self) -> "PoolingParams":
"""Returns a deep copy of the PoolingParams instance.""" """Returns a deep copy of the PoolingParams instance."""
return PoolingParams(additional_data=self.additional_data, ) return PoolingParams(additional_data=self.additional_data)
def __repr__(self) -> str: def __repr__(self) -> str:
return (f"PoolingParams(" return (f"PoolingParams("
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment