From 12d015f0b96f1eef0315519b50ca35781d277f53 Mon Sep 17 00:00:00 2001 From: Guillermo Rodriguez Date: Mon, 20 Jan 2025 00:50:39 -0300 Subject: [PATCH] Fix resp none error and params on flux inpaint pipe --- skynet/dgpu/daemon.py | 2 +- skynet/dgpu/pipes/flux_inpaint.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/skynet/dgpu/daemon.py b/skynet/dgpu/daemon.py index d4fecf3..220fe3c 100644 --- a/skynet/dgpu/daemon.py +++ b/skynet/dgpu/daemon.py @@ -171,7 +171,7 @@ class SkynetDGPUDaemon: logging.info(f'working on {body}') resp = await self.conn.begin_work(rid) - if 'code' in resp: + if not resp or 'code' in resp: logging.info(f'probably being worked on already... skip.') else: diff --git a/skynet/dgpu/pipes/flux_inpaint.py b/skynet/dgpu/pipes/flux_inpaint.py index ba0b9e7..2f88d58 100644 --- a/skynet/dgpu/pipes/flux_inpaint.py +++ b/skynet/dgpu/pipes/flux_inpaint.py @@ -45,6 +45,7 @@ def pipeline_for( torch_dtype=torch.bfloat16, quantization_config=qonfig ) + params['transformer'] = transformer pipe = FluxFillPipeline.from_pretrained( model, **params)