LACAI
/

DialoGPT-small-SGD

Text Generation Transformers PyTorch TensorBoard gpt2 Inference Endpoints text-generation-inference

Model card Files Files and versions Metrics Training metrics Community

Edit model card

YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)

Base model: microsoft/DialoGPT-small

Fine tuned for dialogue response generation on the Schema Guided Dialogue Dataset (Rastogi et al., 2019)

Three additional special tokens were added during the fine-tuning process:

<|pad|> padding token
<|user|> speaker control token to prompt user responses
<|system|> speaker control token to prompt system responses

Downloads last month: 19