What is this

This is AuraFlow pruned to 18 layers from 36 layers, by excising the middle 18 layers. It will make mainly corrupt images until you finetune it for a few thousand steps, after which it should look good.

Usage

pipe = AuraFlowPipeline.from_pretrained("fal/AuraFlow")
del pipe.transformer

transformer = AuraFlowTransformer2DModel.from_pretrained(
    "jimmycarter/auraflow-4.8b")
pipe.transformer = transformer

Downloads last month: 1

Safetensors

Model size

5B params

Tensor type

F32

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support