From 14a2201f7748e939632b18c6990230b11edcc1ee Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Wed, 15 Jun 2022 10:42:37 +0200 Subject: [PATCH] update ldm example --- README.md | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 210efac201..92c74a45ed 100644 --- a/README.md +++ b/README.md @@ -166,13 +166,14 @@ image_pil.save("test.png") #### **Text to Image generation with Latent Diffusion** +_Note: To use latent diffusion install transformers from [this branch](https://github.com/patil-suraj/transformers/tree/ldm-bert)._ + ```python from diffusers import DiffusionPipeline ldm = DiffusionPipeline.from_pretrained("fusing/latent-diffusion-text2im-large") -generator = torch.Generator() -generator = generator.manual_seed(6694729458485568) +generator = torch.manual_seed(42) prompt = "A painting of a squirrel eating a burger" image = ldm([prompt], generator=generator, eta=0.3, guidance_scale=6.0, num_inference_steps=50)