damerajee
/

hathi-moe-test

Text Generation

Mixture of Experts

damerajee/Gaja-v1.00

ai4bharat/Airavata

Model card Files Files and versions

damerajee commited on Mar 29, 2024

Commit

93cdd14

·

verified ·

1 Parent(s): 5fe621b

Update modeling_Hixtral.py

Files changed (1) hide show

modeling_Hixtral.py +1 -1

modeling_Hixtral.py CHANGED Viewed

@@ -1215,7 +1215,7 @@ class HixtralForCausalLM(HixtralPreTrainedModel):
     def get_decoder(self):
         return self.model
-    @add_start_docstrings_to_model_forward(Hixtral_INPUTS_DOCSTRING)
     @replace_return_docstrings(output_type=MoeCausalLMOutputWithPast, config_class=_CONFIG_FOR_DOC)
     # Ignore copy
     def forward(

     def get_decoder(self):
         return self.model
+    @add_start_docstrings_to_model_forward(HIXTRAL_INPUTS_DOCSTRING)
     @replace_return_docstrings(output_type=MoeCausalLMOutputWithPast, config_class=_CONFIG_FOR_DOC)
     # Ignore copy
     def forward(