wanghaofan's picture
Update README.md
7997454 verified
|
raw
history blame
1.16 kB
metadata
license: other
license_name: flux-1-dev-non-commercial-license
license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md
language:
  - en
base_model:
  - black-forest-labs/FLUX.1-dev
library_name: diffusers

Model Details

This is a diffusers version of nyanko7/flux-dev-de-distill.

Usage

import torch
from diffusers import FluxTransformer2DModel
from pipeline_flux_de_distill import FluxPipeline

model_path = "black-forest-labs/FLUX.1-dev"
    
transformer = FluxTransformer2DModel.from_pretrained(
    "InstantX/flux-dev-de-distill-diffusers",
    torch_dtype=torch.bfloat16
)

pipeline = FluxPipeline.from_pretrained(model_path, transformer=transformer, torch_dtype=torch.bfloat16).to("cuda")

prompt = "a tiny astronaut hatching from an egg on the moon"
negative_prompt = "bad photo"

image = pipeline(
    prompt=prompt,
    negative_prompt=negative_prompt,
    guidance_scale=3.5,
    num_inference_steps=24,
).images[0]

image.save("de-distill.jpg")

Other Resources