Update README.md

README.md

@@ -16,4 +16,14 @@ pipe = PixArtAlphaPipeline.from_pretrained("./models--PixArt-alpha--PixArt-XL-2-
                                            torch_dtype=torch.float16, use_safetensors=True)
 
 pipe = pipe.to('cuda')
+pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
+pipe.scheduler.config.prediction_type = "v_prediction"
+generator = torch.manual_seed(318)
+imgs = pipe(prompt="Cute small corgi sitting in a movie theater eating popcorn, unreal engine, 8k.",
+                    num_inference_steps=1, 
+                    num_images_per_prompt = 1,
+                    generator = generator,
+                    guidance_scale=1.,
+                   )[0]
+imgs[0]
 ```