redrafter_model

Redrafter model to support redrafter decoding.

Classes

Base Redrafter Model.

class RedrafterModel

Bases: DynamicModule

Base Redrafter Model.

modify(redrafter_predict_n_tokens=0, redrafter_num_layers=0): Base Redrafter Model modify function. Child class should implement the details.