Update cognitivess_model/modeling_cognitivess.py

cognitivess_model/modeling_cognitivess.py

@@ -223,7 +223,7 @@ class CognitivessForQuestionAnswering(LlamaPreTrainedModel):
         end_logits = end_logits.squeeze(-1).contiguous()
 
         loss = None
-        if start_positions is not None and end_positions are not None:
+        if start_positions is not None and end_positions is not None:
             if len(start_positions.size()) > 1:
                 start_positions = start_positions.squeeze(-1)
             if len(end_positions.size()) > 1: