conversational

Chizuru Ichinose as a DialoGPT model

This model is a fine-tuned version of DialoGPT-medium trained on the Chizuru Ichinose conversational dataset.

We recommend using one of the Transformers pre-built pipelines to keep context without too much work when running this or any DialoGPT model:

from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, Conversational

# load tokenizer and model
tokenizer = AutoTokenizer.from_pretrained("alexandreteles/GPTChizuru")
model = AutoModelForCausalLM.from_pretrained("alexandreteles/GPTChizuru")

# create pipeline
pipe = pipeline(task="conversational", model=model, tokenizer=tokenizer)

# generate response
print(pipe(Conversation("How are you?")))