Upload modeling_fastesm.py with huggingface_hub
Browse files- modeling_fastesm.py +0 -5
modeling_fastesm.py
CHANGED
|
@@ -757,11 +757,6 @@ class FastEsmPreTrainedModel(PreTrainedModel):
|
|
| 757 |
tokenizer = EsmTokenizer.from_pretrained("facebook/esm2_t6_8M_UR50D")
|
| 758 |
all_tied_weights_keys = {}
|
| 759 |
|
| 760 |
-
@classmethod
|
| 761 |
-
def is_remote_code(cls) -> bool:
|
| 762 |
-
# Prevent post-load reinitialization of tensors already loaded from checkpoints.
|
| 763 |
-
return True
|
| 764 |
-
|
| 765 |
@torch.no_grad()
|
| 766 |
def _init_weights(self, module):
|
| 767 |
"""Initialize the weights"""
|
|
|
|
| 757 |
tokenizer = EsmTokenizer.from_pretrained("facebook/esm2_t6_8M_UR50D")
|
| 758 |
all_tied_weights_keys = {}
|
| 759 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 760 |
@torch.no_grad()
|
| 761 |
def _init_weights(self, module):
|
| 762 |
"""Initialize the weights"""
|