                if hidden_states is not None:
                    hidden_states_tensor = torch.from_numpy(hidden_states)
                    predictions = self.model(data_tensor, hidden_states_tensor)
                else:
                    predictions = self.model(data_tensor)