From 38f3c0703f090d6adf074ac5af7f8aafd9365907 Mon Sep 17 00:00:00 2001
From: Adam <24621027+WhiteDopeOnPunk@users.noreply.github.com>
Date: Fri, 30 Sep 2022 03:00:49 -0400
Subject: [PATCH] a

---
 lulzbot.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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