File size: 603 Bytes
ddd56e8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
# 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
''' |