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

add typing indicator

Browse source
This commit is contained in:
Adam 2022-09-29 00:39:28 -04:00
parent 8b675335c6
commit 4aba6be99e
1 changed files with 18 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
lulzbot.py
View file

@ -625,6 +625,24 @@ async def on_message(message):
if message.author == client.user: if message.author == client.user:
return return
if message.channel.name == 'cartman':
async with message.channel.typing():
new_user_input_ids = tokenizer.encode(user_message + tokenizer.eos_token, return_tensors='pt')
bot_output = new_user_input_ids
bot_input_ids = torch.cat([new_user_input_ids, bot_output])
bot_output = model.generate(
bot_input_ids, max_length= 200,
pad_token_id=tokenizer.eos_token_id,
no_repeat_ngram_size=3,
do_sample=True,
top_k=100,
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
if message.channel.name == 'shitposting': if message.channel.name == 'shitposting':
if user_message.lower().count('musk') > 0: if user_message.lower().count('musk') > 0:
await message.channel.send(user_tweet('elonmusk')) await message.channel.send(user_tweet('elonmusk'))
@ -714,21 +732,4 @@ I was finally there\n\
To sit on my throne as the Prince of Bel-Air') To sit on my throne as the Prince of Bel-Air')
return return
if message.channel.name == 'cartman':
new_user_input_ids = tokenizer.encode(user_message + tokenizer.eos_token, return_tensors='pt')
bot_output = new_user_input_ids
bot_input_ids = torch.cat([new_user_input_ids, bot_output])
bot_output = model.generate(
bot_input_ids, max_length= 200,
pad_token_id=tokenizer.eos_token_id,
no_repeat_ngram_size=3,
do_sample=True,
top_k=100,
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
client.run(TOKEN) client.run(TOKEN)