chandar-lab
/

AMPLIFY_120M_base

Model card Files Files and versions

qfournier commited on Oct 2, 2024

Commit

b38fe51

·

verified ·

1 Parent(s): c80a9f3

Update amplify.py

Files changed (1) hide show

amplify.py +1 -1

amplify.py CHANGED Viewed

@@ -203,7 +203,7 @@ class AMPLIFY(AMPLIFYPreTrainedModel):
         # Initialize weights and apply final processing
         self.post_init()
-    def forward(self, input_ids, attention_mask=None, output_hidden_states=False, output_attentions=False):
         # Initialize
         hidden_states, attentions = [], []

         # Initialize weights and apply final processing
         self.post_init()
+    def forward(self, input_ids, attention_mask=None, output_hidden_states=False, output_attentions=False, **kwargs):
         # Initialize
         hidden_states, attentions = [], []