Spaces:

PlayfulTechnology
/

QARAC

Build error

PeteBleackley commited on Oct 9, 2023

Commit

b5ce6f8

1 Parent(s): a0c9643

Unpack BatchEncoding

Files changed (1) hide show

qarac/models/QaracEncoderModel.py CHANGED Viewed

@@ -47,7 +47,8 @@ class QaracEncoderModel(transformers.PreTrainedModel):
             Vector representing the document
         """
         return self.head(self.encoder(input_ids,
                                       attention_mask).last_hidden_state,
                          attention_mask)

             Vector representing the document
         """
+        if attention_mask is None and 'attention_mask' in input_ids:
+            (input_ids,attention_mask) = (input_ids['input_ids'],input_ids['attention_mask'])
         return self.head(self.encoder(input_ids,
                                       attention_mask).last_hidden_state,
                          attention_mask)