diff --git a/skynet/constants.py b/skynet/constants.py index 37ccefd..743270f 100755 --- a/skynet/constants.py +++ b/skynet/constants.py @@ -8,6 +8,7 @@ MODELS = { 'prompthero/openjourney': { 'short': 'midj'}, 'runwayml/stable-diffusion-v1-5': { 'short': 'stable'}, 'stabilityai/stable-diffusion-2-1-base': { 'short': 'stable2'}, + 'snowkidy/stable-diffusion-xl-base-0.9': { 'short': 'stablexl0.9'}, 'stabilityai/stable-diffusion-xl-base-1.0': { 'short': 'stablexl'}, 'Linaqruf/anything-v3.0': { 'short': 'hdanime'}, 'hakurei/waifu-diffusion': { 'short': 'waifu'}, diff --git a/skynet/utils.py b/skynet/utils.py index 2dce173..2ec7a6c 100755 --- a/skynet/utils.py +++ b/skynet/utils.py @@ -82,7 +82,8 @@ def pipeline_for(model: str, mem_fraction: float = 1.0, image=False) -> Diffusio if model == 'runwayml/stable-diffusion-v1-5': params['revision'] = 'fp16' - if model == 'stabilityai/stable-diffusion-xl-base-1.0': + if (model == 'stabilityai/stable-diffusion-xl-base-1.0' or + model == 'snowkidy/stable-diffusion-xl-base-0.9'): if image: pipe_class = StableDiffusionXLImg2ImgPipeline else: