Update README.md

README.md

@@ -61,7 +61,7 @@ Then you can load this model and run inference.
 from sentence_transformers.sparse_encoder import SparseEncoder
 
 # Download from the 🤗 Hub
-model = SparseEncoder("opensearch-project/opensearch-neural-sparse-encoding-doc-v3-gte", trust_remote_code=True)
+model = SparseEncoder("opensearch-project/opensearch-neural-sparse-encoding-doc-v3-gte", trust_remote_code=True, model_kwargs={"code_revision": "40ced75c3017eb27626c9d4ea981bde21a2662f4"})
 
 query = "What's the weather in ny now?"
 document = "Currently New York is rainy."
@@ -135,7 +135,7 @@ def get_tokenizer_idf(tokenizer):
     return torch.tensor(idf_vector)
 
 # load the model
-model = AutoModelForMaskedLM.from_pretrained("opensearch-project/opensearch-neural-sparse-encoding-doc-v3-gte", trust_remote_code=True)
+model = AutoModelForMaskedLM.from_pretrained("opensearch-project/opensearch-neural-sparse-encoding-doc-v3-gte", trust_remote_code=True, code_revision="40ced75c3017eb27626c9d4ea981bde21a2662f4")
 tokenizer = AutoTokenizer.from_pretrained("opensearch-project/opensearch-neural-sparse-encoding-doc-v3-gte")
 idf = get_tokenizer_idf(tokenizer)