From 08da0681cd317f1a0b415f20fb58fb8096db0ce7 Mon Sep 17 00:00:00 2001 From: Konstantine Tsafatinos Date: Wed, 19 Jul 2023 16:14:03 -0400 Subject: [PATCH] add param for stablexl --- skynet/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/skynet/utils.py b/skynet/utils.py index 177f62b..ad13a15 100644 --- a/skynet/utils.py +++ b/skynet/utils.py @@ -79,7 +79,7 @@ def pipeline_for(model: str, mem_fraction: float = 1.0, image=False) -> Diffusio if model == 'snowkidy/stable-diffusion-xl-base-0.9': # TODO: figure out what this does - params['addition_embed_type'] = None + params['addition_embed_type'] = { 'text_time': None } if model == 'runwayml/stable-diffusion-v1-5': params['revision'] = 'fp16'