Apppublic

VisionLanguageGroup/MicroscopyMatching

sourceHugging Faceupdated 2mo agoView on Hugging Face

0likes

load_models.py13 linesDownload Raw Back to _utils

1from config import RunConfig2import torch3from diffusers.pipelines.stable_diffusion import StableDiffusionPipeline4import torch.nn as nn5 6def load_stable_diffusion_model(config: RunConfig):7    device = torch.device('cpu')8 9    stable_diffusion_version = "CompVis/stable-diffusion-v1-4"10    stable = StableDiffusionPipeline.from_pretrained(stable_diffusion_version).to(device)11    return stable12 13