tmp: comment out precision arg

AnswerDotAI · Feb 15, 2024 · 92d99d6 · 92d99d6
1 parent 0106580
commit 92d99d6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ragatouille/models/colbert.py b/ragatouille/models/colbert.py
@@ -11,7 +11,6 @@
 from colbert import Indexer, IndexUpdater, Searcher, Trainer
 from colbert.infra import ColBERTConfig, Run, RunConfig
 from colbert.modeling.checkpoint import Checkpoint
-
 from ragatouille.models.base import LateInteractionModel
 
 # TODO: Move all bsize related calcs to `_set_bsize()`
@@ -485,7 +484,8 @@ def index(
                 documents=None,
                 max_tokens=max_document_length,
             )
-            self._full_vectors_index(half_precision=full_vector_half_precision)
+            self._full_vectors_index()
+            # half_precision=full_vector_half_precision)
         elif self.index_type == "PLAID":
             self._plaid_index(overwrite=overwrite)