suayptalha commited on
Commit
1cdce9a
·
verified ·
1 Parent(s): 659ef62

Update modeling_minGRU.py

Browse files
Files changed (1) hide show
  1. modeling_minGRU.py +2 -3
modeling_minGRU.py CHANGED
@@ -95,11 +95,9 @@ class MinGRUForSequenceClassification(PreTrainedModel):
95
  logits=logits,
96
  )
97
 
 
98
  @classmethod
99
  def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
100
- """
101
- Load model from a pretrained checkpoint.
102
- """
103
  model = super().from_pretrained(pretrained_model_name_or_path, *model_args, **kwargs)
104
 
105
  for name, param in model.named_parameters():
@@ -109,6 +107,7 @@ class MinGRUForSequenceClassification(PreTrainedModel):
109
  print(f"Initialized parameter {name} manually.")
110
 
111
  return model
 
112
 
113
  def save_pretrained(self, save_directory, safe_serialization: Optional[bool] = True, **kwargs):
114
  """
 
95
  logits=logits,
96
  )
97
 
98
+ """
99
  @classmethod
100
  def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
 
 
 
101
  model = super().from_pretrained(pretrained_model_name_or_path, *model_args, **kwargs)
102
 
103
  for name, param in model.named_parameters():
 
107
  print(f"Initialized parameter {name} manually.")
108
 
109
  return model
110
+ """
111
 
112
  def save_pretrained(self, save_directory, safe_serialization: Optional[bool] = True, **kwargs):
113
  """