资源算法waifu_gpt2

waifu_gpt2

2020-03-03 | |  48 |   0 |   0

waifu_gpt2

GPT2 for natural conversation with waifu

This repository demonstrates how to use GPT2 to generate natural conversation with AI waifu. Arbitrary personality can be put into the waifu, just by changing the personality conveyed through the initial dialog.

This code is optimized for waifu conversation. It works extremely fast. The following are measured speeds for obtaining a 4 word response (response time is directly and linearly proportional to number of words in response):

ModelHardwareResponse time (seconds)
distilgpt22 CPU2.22
distilgpt2K80 GPU1.24
distilgpt2T4 GPU0.84
distilgpt2V100 GPU0.80

This is an example initial dialog:

You said: "Rem I missed you so much!"
I said: "I missed you too Subaru!"
You said: "Our date was so fun too"
I said: "Lets get married and be happy!"
You said: "Yes that would be awesome!"
I said: "I like hanging out with you because its so much fun! "
You said: "Lets be happy forever!"

The waifu is then able to have conversation with users.


上一篇:gpt2-keras

下一篇:gpt2-slackbot

用户评价
全部评价

热门资源

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • inferno-boilerplate

    This is a very basic boilerplate example for pe...