Add source id to self.device

pipeline.py

@@ -144,7 +144,7 @@ class RepoEmbeddingPipeline(Pipeline):
             + [tokenizer.sep_token]
         )
         tokens_id = tokenizer.convert_tokens_to_ids(tokens)
-        source_ids = torch.tensor([tokens_id])
+        source_ids = torch.tensor([tokens_id]).to(self.device)
 
         token_embeddings = self.model(source_ids)[0]
         sentence_embeddings = token_embeddings.mean(dim=1)