How to use ostris/Flex.1-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("ostris/Flex.1-alpha", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0]
Hello how many vram we need for this model? it's like flux dev requirements?
No, it's pruned like flux lite so memory use is less than flux dev.Quantized it works fine with 4GB VRAM and 8GB RAM (plus some swap space).
· Sign up or log in to comment
