fix

modeling_jais.py

@@ -813,17 +813,14 @@ class JAISModel(JAISPreTrainedModel):
         if position_ids is not None:
             position_ids = position_ids.view(-1, input_shape[-1])
 
-        import pdb;pdb.set_trace()
         if past_key_values is None:
             past_length = 0
             past_key_values = tuple([None] * len(self.h))
         else:
             if isinstance(past_key_values, tuple):
-                import pdb;pdb.set_trace()
                 past_length = past_key_values[0][0].size(-2)
             else:
                 past_length = past_key_values.get_seq_length()
-            #past_length = past_key_values[0][0].size(-2)
         if position_ids is None:
             position_ids = torch.arange(past_length, input_shape[-1] + past_length, dtype=torch.long, device=device)
             position_ids = position_ids.unsqueeze(0).view(-1, input_shape[-1])