pinecone
/

ConstBERT

Feature Extraction

text-embeddings-inference

Model card Files Files and versions Community

caesar-one commited on 18 days ago

Commit

a6be4c2

·

verified ·

1 Parent(s): 96f6622

Upload ConstBERT

Files changed (2) hide show

modeling.py +2 -2
tokenization_utils.py +1 -1

modeling.py CHANGED Viewed

@@ -3,8 +3,8 @@ from transformers import BertPreTrainedModel, BertModel, AutoTokenizer
 import torch
 from tqdm import tqdm
 from transformers import AutoTokenizer
-from constbert.colbert_configuration import ColBERTConfig
-from constbert.tokenization_utils import QueryTokenizer, DocTokenizer
 class NullContextManager(object):

 import torch
 from tqdm import tqdm
 from transformers import AutoTokenizer
+from .colbert_configuration import ColBERTConfig
+from .tokenization_utils import QueryTokenizer, DocTokenizer
 class NullContextManager(object):

tokenization_utils.py CHANGED Viewed

@@ -1,5 +1,5 @@
 import torch
-from constbert.colbert_configuration import ColBERTConfig
 from transformers import AutoTokenizer
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 import torch
+from .colbert_configuration import ColBERTConfig
 from transformers import AutoTokenizer
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")