dflash_model#

DFlash model to support block-wise parallel speculative decoding.

Classes

DFlashModel

Base DFlash Model.

class DFlashModel#

Bases: DynamicModule

Base DFlash Model.

modify(config)#

Base DFlash Model modify function. Child class should implement the details.