Update esm_scripts/extract.py
Browse files- esm_scripts/extract.py +1 -1
esm_scripts/extract.py
CHANGED
@@ -131,7 +131,7 @@ def run(args):
|
|
131 |
)
|
132 |
|
133 |
|
134 |
-
def run_demo(protein_name, protein_seq, model_location, include, nogpu,
|
135 |
repr_layers=-1, truncation_seq_length=1022, toks_per_batch=4096):
|
136 |
model, alphabet = pretrained.load_model_and_alphabet(model_location)
|
137 |
model.eval()
|
|
|
131 |
)
|
132 |
|
133 |
|
134 |
+
def run_demo(protein_name, protein_seq, model_location, include, nogpu=False,
|
135 |
repr_layers=-1, truncation_seq_length=1022, toks_per_batch=4096):
|
136 |
model, alphabet = pretrained.load_model_and_alphabet(model_location)
|
137 |
model.eval()
|