Update raven_modeling_minimal.py

raven_modeling_minimal.py

@@ -300,7 +300,8 @@ class RavenForCausalLM(RavenPreTrainedModel):
         },
         use_cache: bool = False,
         cache_position: Optional[torch.Tensor] = None,
-    ) -> dict[str, Optional[torch.Tensor]]:
+        **kwargs,
+    ) -> CausalLMOutputRecurrentLatents:
         if position_ids is None and cache_position is None:
             freqs_cis = self.freqs_cis[:, : input_ids.shape[1]]
         elif position_ids is not None: