Views
No views yet
| 1024 x 1792 (4 steps) | |
|---|---|
![]() | |
| bikini model at sea |
1from diffusers import FluxPipeline
2import torch
3import numpy as np
4
5MAX_SEED = np.iinfo(np.int32).max
6seed = random.randint(0, MAX_SEED)
7generator = torch.Generator().manual_seed(seed)
8
9pipeline = FluxPipeline.from_pretrained(
10 "aifeifei798/DarkIdol-1.0", torch_dtype=torch.bfloat16
11).to("cuda")
12
13# Enable VAE big pic
14pipeline.vae.enable_slicing()
15pipeline.vae.enable_tiling()
16
17image = pipeline(
18 prompt="bikini model at sea",
19 guidance_scale=0,
20 num_inference_steps=4,
21 height=1792,
22 width=1024,
23 max_sequence_length=512,
24 generator=generator,
25).images[0]
26image.save("DarkIdol.png")