| # You Only Sample Once (YOSO) | |
| ## Usage | |
| ### 2 step inference | |
| '''python | |
| pipeline = DiffusionPipeline.from_pretrained("stablediffusionapi/realistic-vision-v51", torch_dtype = torch.float16) | |
| pipeline = pipeline.to('cuda') | |
| pipeline.scheduler = LCMScheduler.from_config(pipeline.scheduler.config) | |
| generator = torch.manual_seed(318) | |
| steps = 2 | |
| imgs= pipeline(prompt="A photo of a man, XT3", | |
| num_inference_steps=steps, | |
| num_images_per_prompt = 1, | |
| generator = generator, | |
| guidance_scale=1.5, | |
| )[0] | |
| imgs | |
| ''' |