damerajee commited on
Commit
5c22068
1 Parent(s): 4ebe432

Update modeling_Llamoe.py

Browse files
Files changed (1) hide show
  1. modeling_Llamoe.py +1 -1
modeling_Llamoe.py CHANGED
@@ -916,7 +916,7 @@ class LlamoeModel(LlamoePreTrainedModel):
916
  def set_input_embeddings(self, value):
917
  self.embed_tokens = value
918
 
919
- @add_start_docstrings_to_model_forward(Llamoe_INPUTS_DOCSTRING)
920
  def forward(
921
  self,
922
  input_ids: torch.LongTensor = None,
 
916
  def set_input_embeddings(self, value):
917
  self.embed_tokens = value
918
 
919
+ @add_start_docstrings_to_model_forward(LLAMOE_INPUTS_DOCSTRING)
920
  def forward(
921
  self,
922
  input_ids: torch.LongTensor = None,