Repository: facebookresearch/esm (language: python)

Issue:
Missing model.eval() in main README
First, thanks for making these models available and easily accessible! 
The "quick start" example in the main README is missing a call to model.eval().


Discussion / hints:
I just searched through the closed issues, and I see that was intentional. Might still be good to clarify on the README. Otherwise, feel free to close this issue! Thanks! 
Thanks for flagging, yes it has lead to confusion and worth including even in the minimal quickstart example.

Fix patch:
diff --git a/README.md b/README.md
index d3cccfbd..d41697d8 100644
--- a/README.md
+++ b/README.md
@@ -220,6 +220,7 @@ import esm
 # Load ESM-1b model
 model, alphabet = esm.pretrained.esm1b_t33_650M_UR50S()
 batch_converter = alphabet.get_batch_converter()
+model.eval()  # disables dropout for deterministic results
 
 # Prepare data (first 2 sequences from ESMStructuralSplitDataset superfamily / 4)
 data = [


Test patch:
diff --git a/tests/test_readme.py b/tests/test_readme.py
index 99f6ee90..3a9540b0 100644
--- a/tests/test_readme.py
+++ b/tests/test_readme.py
@@ -26,6 +26,7 @@ def test_readme_2():
     # Load ESM-1b model
     model, alphabet = esm.pretrained.esm1b_t33_650M_UR50S()
     batch_converter = alphabet.get_batch_converter()
+    model.eval()  # disables dropout for deterministic results
 
     # Prepare data (first 2 sequences from ESMStructuralSplitDataset superfamily / 4)
     data = [