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.