search

OpenBMB / ModelCenter

Efficient, Low-Resource, Distributed transformer implementation based on BMTrain
https://modelcenter.readthedocs.io
Apache License 2.0
233 stars 28 forks source link

[FEATURE] generate method #34

Closed h-peng17 closed 2 years ago

h-peng17 commented 2 years ago

Is your feature request related to a problem? Please describe. The decoder (e.g., GPT2) or encoder-decoder models (e.g., T5) needs a generate method to generate text.

Describe the solution you'd like Add generate method in BaseModel.