chore: changed mask token id and pad token id to be int

qdrant · Nov 13, 2024 · d9f6b64 · d9f6b64
1 parent d0db6d8
commit d9f6b64
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: