wenkai commited on
Commit
97b5d9b
1 Parent(s): 52ea231

Update esm_scripts/extract.py

Browse files
Files changed (1) hide show
  1. 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()