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)