Skip to content

Commit f57a2e0

Browse files
authored
Fix Imagic example (#1520)
fix typo, remove incorrect arguments from .train()
1 parent 3ceaa28 commit f57a2e0

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

examples/community/README.md

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -411,7 +411,7 @@ pipe = DiffusionPipeline.from_pretrained(
411411
custom_pipeline="imagic_stable_diffusion",
412412
scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
413413
).to(device)
414-
generator = th.Generator("cuda").manual_seed(0)
414+
generator = torch.Generator("cuda").manual_seed(0)
415415
seed = 0
416416
prompt = "A photo of Barack Obama smiling with a big grin"
417417
url = 'https://www.dropbox.com/s/6tlwzr73jd1r9yk/obama.png?dl=1'
@@ -421,17 +421,15 @@ init_image = init_image.resize((512, 512))
421421
res = pipe.train(
422422
prompt,
423423
image=init_image,
424-
guidance_scale=7.5,
425-
num_inference_steps=50,
426424
generator=generator)
427-
res = pipe(alpha=1)
425+
res = pipe(alpha=1, guidance_scale=7.5, num_inference_steps=50)
428426
os.makedirs("imagic", exist_ok=True)
429427
image = res.images[0]
430428
image.save('./imagic/imagic_image_alpha_1.png')
431-
res = pipe(alpha=1.5)
429+
res = pipe(alpha=1.5, guidance_scale=7.5, num_inference_steps=50)
432430
image = res.images[0]
433431
image.save('./imagic/imagic_image_alpha_1_5.png')
434-
res = pipe(alpha=2)
432+
res = pipe(alpha=2, guidance_scale=7.5, num_inference_steps=50)
435433
image = res.images[0]
436434
image.save('./imagic/imagic_image_alpha_2.png')
437435
```

0 commit comments

Comments
 (0)