medusa

Modules

modelopt.torch.speculative.medusa.conversion

Medusa conversion/restore utilities.

modelopt.torch.speculative.medusa.medusa_model

Medusa model to support medusa decoding.

Medusa Optimization Method.