aifeifei798/DarkIdol-Flux-Photo

Sample

• Art Gallery
• Sample 1
• Sample 2
• step >= 4

from diffusers import FluxPipeline, FluxTransformer2DModel
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)


drakidol_transformer = FluxTransformer2DModel.from_pretrained(
    "aifeifei798/DarkIdol-Flux-Photo",
    subfolder="transformer",
    torch_dtype=torch.float16,
)

pipeline = FluxPipeline.from_pretrained(
    "black-forest-labs/FLUX.1-dev",
    transformer=drakidol_transformer,
    torch_dtype=torch.float16,
)

# Enable VAE big pic
pipeline.vae.enable_slicing()
pipeline.vae.enable_tiling()

image = pipeline(
    prompt="bikini model at sea",
    guidance_scale=0,
    num_inference_steps=6,
    height=1792,
    width=1024,
    generator=generator,
).images[0]
image.save("DarkIdol.png")