LHolten / DialoGPT-MMI-decoder

MMI decoder for DialoGPT and discord bot

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

DialoGPT-MMI-decoder

This is a third party implementation of the MMI-decoder for DialoGPT.

It gives interesting responses, varying between funny, stupid and thoughtfull.

This is how the bot describes itself:

Not sure how to describe it. But the person I was speaking to said it was like a little book of secrets.

Features

  • Unlimitted chat length
  • Discord bot
  • Discord bot tested by a group of ~15 people

Installation

Python requirements:

  • pytorch
  • transformers

Other files

  • Download the medium forward and backward pre-trained model from here
  • Download the config files from here

Place all of these files in the /medium folder

Python requirement for the discord bot:

  • discord.py

Configuration

Modify config.py

Usage

Run interact.py to chat.

Run discord_bot.py to run the discord bot

About

MMI decoder for DialoGPT and discord bot

License:MIT License


Languages

Language:Python 100.0%