Views
No views yet
1from diffusers import FluxPipeline, FluxTransformer2DModel
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
9
10drakidol_transformer = FluxTransformer2DModel.from_pretrained(
11 "aifeifei798/DarkIdol-Flux-Movie",
12 subfolder="transformer",
13 torch_dtype=torch.float16,
14)
15
16pipeline = FluxPipeline.from_pretrained(
17 "black-forest-labs/FLUX.1-dev",
18 transformer=drakidol_transformer,
19 torch_dtype=torch.float16,
20)
21
22# Enable VAE big pic
23pipeline.vae.enable_slicing()
24pipeline.vae.enable_tiling()
25
26image = pipeline(
27 prompt="bikini model at sea",
28 guidance_scale=0,
29 num_inference_steps=6,
30 height=1792,
31 width=1024,
32 generator=generator,
33).images[0]
34image.save("DarkIdol.png")