Update modeling_Llamoe.py
Browse files- 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(
|
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,
|