Update raven_modeling_minimal.py

raven_modeling_minimal.py

@@ -438,7 +438,7 @@ class RavenForCausalLM(RavenPreTrainedModel):
     ):
         model_inputs = {}
         model_inputs["cache_position"] = cache_position
-        current_input_length = model_inputs["input_ids"].shape[1]
+        current_input_length = input_ids.shape[1]
         if past_key_values is not None:
             model_inputs["past_key_values"] = past_key_values
             input_ids = input_ids[:, cache_position]  # type: ignore