remove docstrings

myolmoe/modeling_myolmoe.py

@@ -552,7 +552,7 @@ class OlmoeDecoderLayer(nn.Module):
         return outputs
 
 
-@auto_docstring
+
 class OlmoePreTrainedModel(PreTrainedModel):
     config_class = OlmoeConfig
     base_model_prefix = "model"
@@ -579,7 +579,7 @@ class OlmoePreTrainedModel(PreTrainedModel):
                 module.weight.data[module.padding_idx].zero_()
 
 
-@auto_docstring
+
 class OlmoeModel(OlmoePreTrainedModel):
     def __init__(self, config: OlmoeConfig):
         super().__init__(config)
@@ -605,7 +605,7 @@ class OlmoeModel(OlmoePreTrainedModel):
     def set_input_embeddings(self, value):
         self.embed_tokens = value
 
-    @auto_docstring
+    
     def forward(
         self,
         input_ids: Optional[torch.LongTensor] = None,
@@ -872,7 +872,7 @@ class OlmoeForCausalLM(OlmoePreTrainedModel, GenerationMixin):
     def get_decoder(self):
         return self.model
 
-    @auto_docstring
+    
     def forward(
         self,
         input_ids: Optional[torch.LongTensor] = None,