jimmycarter/auraflow-3.8b

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-3.8b",
    subfolder="transformer")
pipe.transformer = transformer