lhallee commited on
Commit
87d29d9
·
verified ·
1 Parent(s): 8808378

Upload modeling_esm_plusplus.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_esm_plusplus.py +5 -0
modeling_esm_plusplus.py CHANGED
@@ -990,6 +990,11 @@ class PreTrainedESMplusplusModel(PreTrainedModel):
990
  supports_gradient_checkpointing = True
991
  all_tied_weights_keys = {}
992
 
 
 
 
 
 
993
  def _init_weights(self, module):
994
  """Initialize the weights"""
995
  # HF from_pretrained marks loaded parameters with `_is_hf_initialized`.
 
990
  supports_gradient_checkpointing = True
991
  all_tied_weights_keys = {}
992
 
993
+ @classmethod
994
+ def is_remote_code(cls) -> bool:
995
+ # Prevent post-load reinitialization of tensors already loaded from checkpoints.
996
+ return True
997
+
998
  def _init_weights(self, module):
999
  """Initialize the weights"""
1000
  # HF from_pretrained marks loaded parameters with `_is_hf_initialized`.