lhallee commited on
Commit
b67619f
·
verified ·
1 Parent(s): b45f5d5

Upload modeling_fastesm.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_fastesm.py +1 -1
modeling_fastesm.py CHANGED
@@ -559,7 +559,7 @@ class FastEsmPreTrainedModel(PreTrainedModel):
559
  sequences = list(set([seq[:max_len] for seq in sequences]))
560
  sequences = sorted(sequences, key=len, reverse=True)
561
  dataset = ProteinDataset(sequences)
562
- dataloader = DataLoader(dataset, batch_size=batch_size, num_workers=num_workers, collate_fn=self._collate_fn)
563
  device = self.device
564
 
565
  def get_embeddings(residue_embeddings: torch.Tensor, attention_mask: Optional[torch.Tensor] = None) -> torch.Tensor:
 
559
  sequences = list(set([seq[:max_len] for seq in sequences]))
560
  sequences = sorted(sequences, key=len, reverse=True)
561
  dataset = ProteinDataset(sequences)
562
+ dataloader = DataLoader(dataset, batch_size=batch_size, num_workers=num_workers, collate_fn=self._collate_fn, shuffle=False)
563
  device = self.device
564
 
565
  def get_embeddings(residue_embeddings: torch.Tensor, attention_mask: Optional[torch.Tensor] = None) -> torch.Tensor: