diff --git a/lulzbot.py b/lulzbot.py index 799bc87..ab6d026 100644 --- a/lulzbot.py +++ b/lulzbot.py @@ -19,6 +19,7 @@ client = discord.Client(activity=discord.Game(name='with myself')) import numpy as np from transformers import AutoTokenizer, AutoModelForCausalLM import torch +import asyncio tokenizer = AutoTokenizer.from_pretrained('microsoft/DialoGPT-large') model = AutoModelForCausalLM.from_pretrained('../southpark/output-small') @@ -639,7 +640,6 @@ async def on_message(message): top_p=0.7, temperature=.8 ) - await message.channel.send('{}'.format(tokenizer.decode(bot_output[:,bot_input_ids.shape[-1]:][0], skip_special_tokens=True))) return