Fix imports for LSG code: explicit BERT import + import math + logger

README.md

@@ -14,7 +14,7 @@ Below is an example to encode queries and passages from the MS-MARCO passage ran
 
 ```python
 from sentence_transformers import SentenceTransformer
-model = SentenceTransformer('efederici/e5-large-v2-4096', {"trust_remote_code": True})
+model = SentenceTransformer('guymorganb/e5-large-v2-4096-lsg-patched', {"trust_remote_code": True})
 input_texts = [
     'query: how much protein should a female eat',
     'query: summit define',
@@ -45,8 +45,8 @@ input_texts = [
   "passage: Definition of summit for English Language Learners. : 1  the highest point of a mountain : the top of a mountain. : 2  the highest level. : 3  a meeting or series of meetings between the leaders of two or more governments."
 ]
 
-tokenizer = AutoTokenizer.from_pretrained('efederici/e5-large-v2-4096')
-model = AutoModel.from_pretrained('efederici/e5-large-v2-4096', trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained('guymorganb/e5-large-v2-4096-lsg-patched')
+model = AutoModel.from_pretrained('guymorganb/e5-large-v2-4096-lsg-patched', trust_remote_code=True)
 
 batch_dict = tokenizer(input_texts, max_length=4096, padding=True, truncation=True, return_tensors='pt')
 outputs = model(**batch_dict)