From 820c856e55854cc9d7b25f49184c35dd623c6c46 Mon Sep 17 00:00:00 2001 From: Guillermo Rodriguez Date: Sat, 24 Dec 2022 11:46:50 -0300 Subject: [PATCH] Fix image meta and add logging on telgram --- skynet/frontend/telegram.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/skynet/frontend/telegram.py b/skynet/frontend/telegram.py index bf9d116..678e19c 100644 --- a/skynet/frontend/telegram.py +++ b/skynet/frontend/telegram.py @@ -28,6 +28,8 @@ def prepare_metainfo_caption(meta: dict) -> str: meta_str += f'step: {meta["step"]}\n' meta_str += f'guidance: {meta["guidance"]}\n' meta_str += f'algo: \"{meta["algo"]}\"\n' + if meta['upscaler']: + meta_str += f'upscaler: \"{meta["upscaler"]}\"\n' meta_str += f'sampler: k_euler_ancestral\n' meta_str += f'skynet v{VERSION}' return meta_str @@ -101,6 +103,7 @@ async def run_skynet_telegram( else: logging.info(resp.result['id']) img_raw = base64.b64decode(bytes.fromhex(resp.result['img'])) + print(f'got image of size: {len(img_raw)}') size = (512, 512) if resp.result['meta']['upscaler'] == 'x4': size = (2048, 2048)