Update modelling_pisco.py

modelling_pisco.py

@@ -228,7 +228,6 @@ class PISCO(PreTrainedModel):
         questions: list of string
         compressed_documents: torch tensor, its first dimension should be a multiple of len(questions)
         """
-        print(compressed_documents.size(), len(questions))
         self.generation_top_k = compressed_documents.size(0) // len(questions)
         assert compressed_documents.size(0) % self.generation_top_k == 0, f"{compressed_documents.size(0)} {self.generation_top_k}"