DarkIdol-1.0
1024 x 1792 (4 steps) |
|
|
bikini model at sea
|
Inference code
from diffusers import FluxPipeline
import torch
import numpy as np
MAX_SEED = np.iinfo(np.int32).max
seed = random.randint(0, MAX_SEED)
generator = torch.Generator().manual_seed(seed)
pipeline = FluxPipeline.from_pretrained(
"aifeifei798/DarkIdol-1.0", torch_dtype=torch.bfloat16
).to("cuda")
pipeline.vae.enable_slicing()
pipeline.vae.enable_tiling()
image = pipeline(
prompt="bikini model at sea",
guidance_scale=0,
num_inference_steps=4,
height=1792,
width=1024,
max_sequence_length=512,
generator=generator,
).images[0]
image.save("DarkIdol.png")
Documentation