add param for stablexl

pull/11/head
Konstantine Tsafatinos 2023-07-19 16:11:48 -04:00
parent bcb499448e
commit ae348c7c6f
1 changed files with 4 additions and 0 deletions

View File

@ -77,6 +77,10 @@ def pipeline_for(model: str, mem_fraction: float = 1.0, image=False) -> Diffusio
'safety_checker': None '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': if model == 'runwayml/stable-diffusion-v1-5':
params['revision'] = 'fp16' params['revision'] = 'fp16'