Upload modeling_fastesm.py with huggingface_hub
Browse files- 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:
|