feat: Add ColBERT late interaction model support (#33686)
Signed-off-by: Ilya Boytsov <ilyaboytsov1805@gmail.com> Signed-off-by: Ilya Boytsov <boytsovpanamera@mail.ru> Co-authored-by: Cyrus Leung <cyrus.tl.leung@gmail.com> Co-authored-by: wang.yuqi <yuqi.wang@daocloud.io>
This commit is contained in:
@@ -1411,6 +1411,11 @@ class ModelConfig:
|
||||
self._model_info.supports_cross_encoding or self.convert_type == "classify"
|
||||
)
|
||||
|
||||
@property
|
||||
def is_late_interaction(self) -> bool:
|
||||
"""Check if model uses late interaction (ColBERT-style) scoring."""
|
||||
return self._model_info.supports_late_interaction
|
||||
|
||||
@property
|
||||
def is_pp_supported(self) -> bool:
|
||||
return self._model_info.supports_pp
|
||||
|
||||
Reference in New Issue
Block a user