mirror of https://github.com/vladmandic/automatic
18 lines
964 B
Python
18 lines
964 B
Python
import diffusers.pipelines as p
|
|
|
|
|
|
def is_sd15(model):
|
|
if model is None:
|
|
return False
|
|
if hasattr(model, '__name__'):
|
|
return model.__name__ == p.StableDiffusionPipeline.__name__ or model.__name__ == p.StableDiffusionImg2ImgPipeline.__name__ or model.__name__ == p.StableDiffusionInpaintPipeline.__name__
|
|
return isinstance(model, p.StableDiffusionPipeline) or isinstance(model, p.StableDiffusionImg2ImgPipeline) or isinstance(model, p.StableDiffusionInpaintPipeline)
|
|
|
|
|
|
def is_sdxl(model):
|
|
if model is None:
|
|
return False
|
|
if hasattr(model, '__name__'):
|
|
return model.__name__ == p.StableDiffusionXLPipeline.__name__ or model.__name__ == p.StableDiffusionXLImg2ImgPipeline.__name__ or model.__name__ == p.StableDiffusionXLInpaintPipeline.__name__
|
|
return isinstance(model, p.StableDiffusionXLPipeline) or isinstance(model, p.StableDiffusionXLImg2ImgPipeline) or isinstance(model, p.StableDiffusionXLInpaintPipeline)
|