From ae348c7c6fd679cba1475c12e0860a16a888f4b8 Mon Sep 17 00:00:00 2001
From: Konstantine Tsafatinos <kosta.tsaf@gmail.com>
Date: Wed, 19 Jul 2023 16:11:48 -0400
Subject: [PATCH] add param for stablexl

---
 skynet/utils.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/skynet/utils.py b/skynet/utils.py
index 2837118..177f62b 100644
--- a/skynet/utils.py
+++ b/skynet/utils.py
@@ -77,6 +77,10 @@ def pipeline_for(model: str, mem_fraction: float = 1.0, image=False) -> Diffusio
         'safety_checker': None
     }
 
+    if model == 'snowkidy/stable-diffusion-xl-base-0.9':
+        # TODO: figure out what this does
+        params['addition_embed_type'] = None
+
     if model == 'runwayml/stable-diffusion-v1-5':
         params['revision'] = 'fp16'