Apppublic

bala1802/StableDiffusionModel

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face

0likes

model.py16 linesDownload Raw Back to root

1import torch2from diffusers import DiffusionPipeline3import config4 5def initialize_diffusion_model():6    pretrained_model_name_or_path = config.STABLE_DIFFUSION_MODEL7    pipe = DiffusionPipeline.from_pretrained(pretrained_model_name_or_path,8                                             torch_dtype=torch.float16).to(config.DEVICE)9    10    pipe.load_textual_inversion("sd-concepts-library/dreams")11    pipe.load_textual_inversion("sd-concepts-library/midjourney-style") 12    pipe.load_textual_inversion("sd-concepts-library/moebius") 13    pipe.load_textual_inversion("sd-concepts-library/style-of-marc-allante") 14    pipe.load_textual_inversion("sd-concepts-library/wlop-style")15 16    return pipe