fix repetition penalty

modeling_cpmbee.py

@@ -1840,10 +1840,10 @@ class CpmBeeForCausalLM(CpmBeePreTrainedModel):
                 next_token_logits,
                 batch_size,
                 num_beams,
-                model_inputs["input_ids"],
+                input_ids,
                 repetition_penalty,
                 pred_start_index,
-                model_inputs["input_ids"].size(-1) - 1,
+                input_ids.size(-1) - 1,
                 None,
             )