BERT-UNILM-for-Poetry-Gen
This is a Poetry Generation project based on BERT-UNILM. We use the chinese-bert-wwm as pre-trained model. You could get the pre-trained model from here.
For more details, you could visit our repositry in github: BERT-UNILM-for-Poetry-Gen.Welcome to star this repo if you like it!
Environment
- Pytorch
- Datasets: chinese-poetry
Feature
- Generate Chinese poems with different styles
- Strength the
beam_searchmethod - Deploy the model to the web
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support
Model tree for Wink-Snow/BERT-UNILM-for-Poetry-GEN
Base model
hfl/chinese-bert-wwm