From 43ae55077c5c793d382b7ee3ed6b57630bbcd22e Mon Sep 17 00:00:00 2001 From: Omar Khattab Date: Wed, 19 Apr 2023 17:21:22 -0700 Subject: [PATCH] Update base_colbert.py --- colbert/modeling/base_colbert.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/colbert/modeling/base_colbert.py b/colbert/modeling/base_colbert.py index 19084172..12cf68c5 100644 --- a/colbert/modeling/base_colbert.py +++ b/colbert/modeling/base_colbert.py @@ -26,10 +26,12 @@ def __init__(self, name_or_path, colbert_config=None): try: HF_ColBERT = class_factory(self.name) except: - HF_ColBERT = class_factory('bert-base-uncased') + self.name = 'bert-base-uncased' # TODO: Double check that this is appropriate here in all cases + HF_ColBERT = class_factory(self.name) - assert self.name is not None - HF_ColBERT = class_factory(self.name) + # assert self.name is not None + # HF_ColBERT = class_factory(self.name) + self.model = HF_ColBERT.from_pretrained(name_or_path, colbert_config=self.colbert_config) self.raw_tokenizer = AutoTokenizer.from_pretrained(name_or_path)