Instructions to use Freepik/flux.1-lite-8B-alpha with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use Freepik/flux.1-lite-8B-alpha with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("Freepik/flux.1-lite-8B-alpha", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps
- Draw Things
- DiffusionBee
Training Cost
#16
by zhangyang-0123 - opened
Hi,
Would you mind sharing the training hours to finetune the MMDit 4? I am wondering if it is possible for anyone to distill with an affordable budget on cloud.
We trained the final version of (flux-lite-8b) on 8x H200 GPUs for about 140 hours. Plus ~1-2 extra days to pre-compute the input and target hidden states.
Hi, thanks for the awesome project! Could you share about the resolution you used for the distillation? Are you doing it multiple phases (i.e.: 512->1024) or you only do one size or just randomly mixed?