Update src/models/pairwise_model.py

src/models/pairwise_model.py

@@ -14,7 +14,7 @@ pad_token_id = tokenizer.pad_token_id
 
 
 class PairwiseModel(nn.Module):
-    def __init__(self, model_name, max_length=384, batch_size=16, device="cuda:0"):
+    def __init__(self, model_name, max_length=384, batch_size=16, device="cpu"):
         super(PairwiseModel, self).__init__()
         self.max_length = max_length
         self.batch_size = batch_size