From fadb4eab6d9e3b3e7d5d49193dcf67f1a1b65c15 Mon Sep 17 00:00:00 2001 From: Konstantine Tsafatinos Date: Tue, 18 Jul 2023 12:40:21 -0400 Subject: [PATCH] update params for work queue --- skynet/frontend/discord/__init__.py | 10 ++++++---- skynet/frontend/discord/handlers.py | 6 +++++- 2 files changed, 11 insertions(+), 5 deletions(-) diff --git a/skynet/frontend/discord/__init__.py b/skynet/frontend/discord/__init__.py index a9518b3..42afd17 100644 --- a/skynet/frontend/discord/__init__.py +++ b/skynet/frontend/discord/__init__.py @@ -134,6 +134,7 @@ class SkynetDiscordFrontend: }) request_time = datetime.now().isoformat() + # import pdb; pdb.set_trace() # await self.update_status_message( # status_msg, # f'processing a \'{method}\' request by {tg_user_pretty(user)}\n' @@ -154,13 +155,14 @@ class SkynetDiscordFrontend: }, self.account, self.key, permission=self.permission ) + print(res) if 'code' in res or 'statusCode' in res: logging.error(json.dumps(res, indent=4)) - # await self.update_status_message( - # status_msg, - # 'skynet has suffered an internal error trying to fill this request') - # return + await self.bot.send( + status_msg, + 'skynet has suffered an internal error trying to fill this request') + return enqueue_tx_id = res['transaction_id'] enqueue_tx_link = hlink( diff --git a/skynet/frontend/discord/handlers.py b/skynet/frontend/discord/handlers.py index 8ac46a5..c408840 100644 --- a/skynet/frontend/discord/handlers.py +++ b/skynet/frontend/discord/handlers.py @@ -38,8 +38,12 @@ def create_handler_context(frontend: 'SkynetDiscordFrontend'): user = 'tests' status_msg = 'status' params = { - 'prompt': prompt, + 'prompt': arg, + 'seed': None, + 'step': 35, + 'guidance': 1, } + # import pdb; pdb.set_trace() ec = await work_request(user, status_msg, 'txt2img', params) print(ec)