wenkai commited on
Commit
d6d5757
1 Parent(s): 61cedea

Update esm_scripts/extract.py

Browse files
Files changed (1) hide show
  1. esm_scripts/extract.py +1 -1
esm_scripts/extract.py CHANGED
@@ -139,7 +139,7 @@ def run_demo(protein_name, protein_seq, model, alphabet, include,
139
  data_loader = torch.utils.data.DataLoader(
140
  dataset, collate_fn=alphabet.get_batch_converter(truncation_seq_length), batch_sampler=batches
141
  )
142
- print(f"Read {fasta_file} with {len(dataset)} sequences")
143
 
144
  # output_dir.mkdir(parents=True, exist_ok=True)
145
  return_contacts = "contacts" in include
 
139
  data_loader = torch.utils.data.DataLoader(
140
  dataset, collate_fn=alphabet.get_batch_converter(truncation_seq_length), batch_sampler=batches
141
  )
142
+ print(f"Read sequences")
143
 
144
  # output_dir.mkdir(parents=True, exist_ok=True)
145
  return_contacts = "contacts" in include