Downloads · 30 days
100
2% of all-time downloads
CDial-GPT2_LCCC-base
thu-coai/CDial-GPT2_LCCC-base
CDial-GPT2_LCCC-base is a machine learning model from thu-coai. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as mit.
This project provides a large-scale Chinese GPT model pre-trained on the dataset LCCC.
All-time downloads
6.4K
Public
Parameters
98.6M
789 MB on disk
Likes
7
Public
Repo makeup
Click a slice to open those files.
.bin395 MB · 50%
At a glance
- Library
- transformers
- License
- mit
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Sep 8, 2023
- SHA
- c9094991
Datasets
Spaces
Model card
From the Hugging Face model README
Chinese pre-trained dialogue model (CDial-GPT)
This project provides a large-scale Chinese GPT model pre-trained on the dataset LCCC.
We present a series of Chinese GPT model that are first pre-trained on a Chinese novel dataset and then post-trained on our LCCC dataset.
Similar to TransferTransfo, we concatenate all dialogue histories into one context sentence, and use this sentence to predict the response. The input of our model consists of word embedding, speaker embedding, and positional embedding of each word.
Paper: A Large-Scale Chinese Short-Text Conversation Dataset
How to use
from transformers import OpenAIGPTLMHeadModel, GPT2LMHeadModel, BertTokenizer
import torch
tokenizer = BertTokenizer.from_pretrained("thu-coai/CDial-GPT2_LCCC-base")
model = GPT2LMHeadModel.from_pretrained("thu-coai/CDial-GPT2_LCCC-base")
For more details, please refer to our repo. on github.
- Library
- transformers
- License
- mit
- Created
- Mar 2, 2022
- Updated
- Sep 8, 2023