Spaces:

Jet-12138
/

CommentResponse

Runtime error

Jet-12138 commited on Apr 25, 2025

Commit

89386d1

verified ·

1 Parent(s): ff59e33

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -47,7 +47,7 @@ class CommentMTLModel(nn.Module):
         # Pass input through BERT model
         outputs = self.bert(
             input_ids=input_ids,
-            attention_mask=attention_mask
             token_type_ids=token_type_ids
         )

         # Pass input through BERT model
         outputs = self.bert(
             input_ids=input_ids,
+            attention_mask=attention_mask,
             token_type_ids=token_type_ids
         )