metadata
license: apache-2.0
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