Update modelling_pisco.py

modelling_pisco.py

@@ -65,14 +65,6 @@ class PISCO(PreTrainedModel):
         self.adapter_keys.append('encoder_adapter')
         
         self.generation_config = GenerationConfig(do_sample=False, top_p=None)
-        
-        print('a')
-        # self.decoder = self.decoder.to('cuda')
-        print('b')
-        if torch.cuda.is_available():
-            print('c')
-            # self.decoder = self.decoder.to('cuda')
-            print('d')
 
     def create_tokenizer(self, cfg):
         self.tokenizer = AutoTokenizer.from_pretrained(cfg.decoder_model_name, use_fast=True, padding_side='left')
@@ -236,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}"