| 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** | |
| ```py | |
| pipe = AuraFlowPipeline.from_pretrained("fal/AuraFlow") | |
| del pipe.transformer | |
| transformer = AuraFlowTransformer2DModel.from_pretrained( | |
| "jimmycarter/auraflow-3.8b", | |
| subfolder="transformer") | |
| pipe.transformer = transformer | |
| ``` |