Update README.md

README.md

@@ -50,9 +50,8 @@ def load_model_and_tokenizer(model_name: str) -> Tuple[AutoModel, AutoTokenizer]
     """
     model = AutoModelForTextEncoding.from_pretrained(
         model_name, torch_dtype=torch.bfloat16, device_map="auto"
-    )
+    ).eval()
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model.eval()  # Deactivate Dropout
     return model, tokenizer
 ```