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 許可證。