Update RepoPipeline.py

RepoPipeline.py

@@ -157,9 +157,9 @@ class RepoPipeline(Pipeline):
 
     def generate_embeddings(self, text_sets, max_length):
         assert max_length < 1024
-        return torch.concat([self.encode(text, max_length) for text in text_sets], dim=0) \
+        return torch.zeros((1, 768), device=self.device) \
             if text_sets is None or len(text_sets) == 0 \
-            else torch.zeros((1, 768), device=self.device)
+            else torch.cat([self.encode(text, max_length) for text in text_sets], dim=0)
 
     def _forward(self, extracted_infos: List, max_length=512) -> List:
         model_outputs = []