skygpu
/
skynet
mirror of https://github.com/skygpu/skynet.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix resp none error and params on flux inpaint pipe

pull/44/head
Guillermo Rodriguez 2025-01-20 00:50:39 -03:00
parent 48a0d0db79
commit 12d015f0b9
No known key found for this signature in database
GPG Key ID: 002CC5F1E6BDA53E
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
skynet/dgpu/daemon.py
View File

@ -171,7 +171,7 @@ class SkynetDGPUDaemon:
logging.info(f'working on {body}') logging.info(f'working on {body}')
resp = await self.conn.begin_work(rid) 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.') logging.info(f'probably being worked on already... skip.')
else: else:

1
skynet/dgpu/pipes/flux_inpaint.py
View File

@ -45,6 +45,7 @@ def pipeline_for(
torch_dtype=torch.bfloat16, torch_dtype=torch.bfloat16,
quantization_config=qonfig quantization_config=qonfig
) )
params['transformer'] = transformer
pipe = FluxFillPipeline.from_pretrained( pipe = FluxFillPipeline.from_pretrained(
model, **params) model, **params)