Update modeling_progen.py

modeling_progen.py

@@ -613,11 +613,7 @@ class ProGenForCausalLM(ProGenPreTrainedModel):
         # compute loss in fp32 to match with mesh-tf version
         # https://github.com/EleutherAI/gpt-neo/blob/89ce74164da2fb16179106f54e2269b5da8db333/models/gpt2/gpt2.py#L179
         lm_logits = self.lm_head(hidden_states).to(torch.float32)
-        print(f"lm_logits = {lm_logits.shape}")
-        print(f"logits_to_keep = {logits_to_keep}")
 
-        # Debug shape
-        print(f"Final logits shape: {lm_logits.shape}")  # Should be [batch, seq, vocab]
         loss = None
         if labels is not None:
             # Shift so that tokens < n predict n