From d9f6b64b78d260f12e2da7c1e33915e6ec840190 Mon Sep 17 00:00:00 2001 From: hh-space-invader Date: Fri, 8 Nov 2024 15:28:35 +0200 Subject: [PATCH] chore: changed mask token id and pad token id to be int --- fastembed/late_interaction/colbert.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fastembed/late_interaction/colbert.py b/fastembed/late_interaction/colbert.py index 89b87805..92d74c57 100644 --- a/fastembed/late_interaction/colbert.py +++ b/fastembed/late_interaction/colbert.py @@ -175,8 +175,8 @@ def __init__( self._model_dir = self.download_model( self.model_description, self.cache_dir, local_files_only=self._local_files_only ) - self.mask_token_id = None - self.pad_token_id = None + self.mask_token_id = int() + self.pad_token_id = int() self.skip_list = set() if not self.lazy_load: