pub fn load_stable_diffusion<B: Backend>(
    path: &str,
    device: &B::Device
) -> Result<StableDiffusion<B>, Box<dyn Error>>