Auraflow V0.3

🚀 AuraFlow v0.3

AuraFlow v0.3 是一款完全开源的基于流的文本到图像生成模型。与上一版本 AuraFlow-v0.2 相比，该模型在训练时使用了更多的计算资源。与 AuraFlow-v0.2 相比，此模型在更多具有美感的数据集上进行了微调，现在支持多种宽高比（目前宽度和高度最大可达 1536 像素）。

🚀 快速开始

📦 安装指南

$ pip install transformers accelerate protobuf sentencepiece
$ pip install git+https://github.com/huggingface/diffusers.git

💻 使用示例

基础用法

from diffusers import AuraFlowPipeline
import torch

pipeline = AuraFlowPipeline.from_pretrained(
    "terminusresearch/auraflow-v0.3",
    torch_dtype=torch.float16,
    variant="fp16",
).to("cuda")

image = pipeline(
    prompt="rempage of the iguana character riding F1, fast and furious, cinematic movie poster",
    width=1536,
    height=768,
    num_inference_steps=50, 
    generator=torch.Generator().manual_seed(1),
    guidance_scale=3.5,
).images[0]

image.save("output.png")

📄 许可证

本项目采用 Apache-2.0 许可证。