minpeter/FLUX-majicflus-v1-diffusers

majicflus v1

Convert single file diffusers weights to load from_pretrained in a very simple way.

from diffusers import FluxPipeline, FluxTransformer2DModel
import torch

dtype = torch.float8_e4m3fn
transformer = (
    FluxTransformer2DModel.from_single_file(
        # Remove the "model.diffusion_model." prefix from the safetensors key
        "./majicflus_v1_cleaned.safetensors",
        torch_dtype=dtype,
    )
    .to("cuda")
    .to(torch.bfloat16)
)

pipe = FluxPipeline.from_pretrained(
    "black-forest-labs/FLUX.1-dev", transformer=transformer
)

pipe.save_pretrained("./majicflus_v1_conv_diffusers/model")