--- license: other license_name: flux-1-dev-non-commercial-license license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/LICENSE.md base_model: - black-forest-labs/FLUX.1-dev pipeline_tag: text-to-image library_name: diffusers tags: - flux - text-to-image --- ![Flux.1 Lite](./sample_images/flux1-lite-8B_sample.png) # Flux.1 Lite We want to announce the alpha version of our new distilled Flux.1 Lite model, an 8B parameter transformer model distilled from the original Flux1. dev model. Our goal is to further reduce FLUX.1-dev transformer parameters up to 24Gb to make it compatible with most of GPU cards. ![Flux.1 Lite vs FLUX.1-dev](./sample_images/models_comparison.png) ## Text-to-Image Usage It is recommended to use a `guidance_scale` of 3.5 and a `n_steps` between 22 and 30 for best results. ```python import torch from diffusers import FluxPipeline base_model_id = "Freepik/flux.1-lite-8B-alpha" torch_dtype = torch.bfloat16 device = "cuda" # Load the pipe model_id = "Freepik/flux.1-lite-8B-alpha" pipe = FluxPipeline.from_pretrained( model_id, torch_dtype=torch_dtype ).to(device) # Inference prompt = "Scene inspired by 2000 comedy animation, a glowing green alien whose fluorescent skin emits light, standing in a dark purple forest. The alien is holding a large sign that reads 'LITE 8B ALPHA' in bold letters. The forest around is shadowy, with tall, eerie trees and mist rolling in. The alien radiates a soft, supernatural glow, illuminating the surroundings, creating a stark contrast between light and darkness. Style of an old comic, with flat colors, halftone shading, and a slightly weathered, vintage texture." guidance_scale = 3.5 # Important to keep guidance_scale to 3.5 n_steps = 28 seed = 11 with torch.inference_mode(): image = pipe( prompt=prompt, generator=torch.Generator(device="cpu").manual_seed(seed), num_inference_steps=n_steps, guidance_scale=guidance_scale, height=1024, width=1024, ).images[0] image.save("output.png") ``` ## Checkpoints * `flux.1-lite-8B-alpha.safetensors`: Transformer checkpoint, in Flux original format. * `transformers/`: Contains distilled 8B transformer model, in diffusers format. ## Try our Hugging Face demos: Flux.1 Lite demo host on [πŸ€— flux.1-lite](https://huggingface.co/spaces/Freepik/flux.1-lite) ## NewsπŸ”₯πŸ”₯πŸ”₯ * Oct.18, 2024. Alpha 8B checkpoint and comparison demo πŸ€— (i.e. [Flux.1 Lite](https://huggingface.co/spaces/Freepik/flux.1-lite)) is publicly available on [HuggingFace Repo](https://huggingface.co/Freepik/flux.1-lite-8B-alpha).