import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("aifeifei798/DarkIdol-Flux-Fashion", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]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-Fashion",
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")
- Downloads last month
- -
Model tree for aifeifei798/DarkIdol-Flux-Fashion
Base model
black-forest-labs/FLUX.1-dev Finetuned
black-forest-labs/FLUX.1-Krea-dev