# Image cannot run correctly on stable-diffusion-2-1

**URL:** <https://discuss.huggingface.co/t/image-cannot-run-correctly-on-stable-diffusion-2-1/46528>\
**Category:** 🧨 Diffusers\
**Created:** [July 12, 2023, 5:43am UTC](https://discuss.huggingface.co/t/image-cannot-run-correctly-on-stable-diffusion-2-1/46528 "2023-07-12T05:43:43Z")\
**Posts on this page:** 1\
**Page:** 1

<div class="post-metadata">

**Author:** ![fengkuanglzc](https://avatars.discourse-cdn.com/v4/letter/f/82dd89/32.png) [@fengkuanglzc](https://discuss.huggingface.co/u/fengkuanglzc)\
**Post date:** [July 12, 2023, 5:43am UTC](https://discuss.huggingface.co/t/image-cannot-run-correctly-on-stable-diffusion-2-1/46528/1 "2023-07-12T05:43:43Z")

</div>

```auto
import requests
# import ptp_utils
from PIL import Image
from io import BytesIO
import torch
import os
from diffusers import DiffusionPipeline, DDIMScheduler, DPMSolverMultistepScheduler
has_cuda = torch.cuda.is_available()
device = torch.device('cpu' if not has_cuda else 'cuda:4')
pipe = DiffusionPipeline.from_pretrained(
    "./models/stable-diffusion-2-1",
        safety_checker=None,
    use_auth_token=True,
    custom_pipeline="imagic_stable_diffusion",
).to(device)
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)

generator = torch.Generator("cuda").manual_seed(1)
seed = 0
prompt = "A laughing man"
url = './images/p1.jpeg'
init_image = Image.open(url).convert("RGB")
init_image = init_image.resize((768, 768))

res = pipe.train(
    prompt,
    image=init_image,
    text_embedding_optimization_steps = 1000,
    embedding_learning_rate = 2e-3,
    model_fine_tuning_optimization_steps = 1500,
    diffusion_model_learning_rate = 5e-7,
    generator=generator)
res = pipe(alpha=0.3, guidance_scale=7.5, num_inference_steps=50)
os.makedirs("imagic", exist_ok=True)
image = res.images[0]

```

I test [Imagic](https://github.com/huggingface/diffusers/blob/main/examples/community/imagic_stable_diffusion.py) on stable-diffusion-2-1, but I cannot got a reasonable output. The same code I have test on stable-diffusion-v1-5 which works well.

 ![download](https://us1.discourse-cdn.com/hellohellohello/original/3X/f/a/fa405bf62da80cfb73ba8445e631fc8c5223c50f.jpeg)