jinze

Hello authors, While reading your code, I noticed that the multiple Medusa Heads you proposed are computing results in parallel ``` for i in range(self.medusa): medusa_logits.append(self.medusa_head[i](hidden_states)) ``` (although the later...