(self, hidden_states)
| 1077 | self.model.empty_input_forward(self.forward_meta) |
| 1078 | |
| 1079 | def _get_self_hidden_states(self, hidden_states): |
| 1080 | target_hidden_states = eagle_get_self_hidden_states( |
| 1081 | hidden_states, |
| 1082 | self.model_inputs.last_seq_lens_this_time, |
| 1083 | self.model_inputs["seq_lens_this_time"], |
| 1084 | self.model_inputs["step_idx"], |
| 1085 | ) |
| 1086 | self.model_inputs["target_hidden_states"].copy_(target_hidden_states, False) |
| 1087 | |
| 1088 | def update_task_chunk_prefill(self, task): |
| 1089 | """ |
no outgoing calls
no test coverage detected