adam/lulzbot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

a

Browse source
This commit is contained in:
Adam 2022-09-30 03:00:49 -04:00
parent 4aba6be99e
commit 38f3c0703f
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lulzbot.py
View file

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