许可证：apache-2.0
标签：

• 文本生成图像

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="蜥蜴角色骑F1赛车的电影海报，速度与激情，电影感十足",  
    width=1536,  
    height=768,  
    num_inference_steps=50,  
    generator=torch.Generator().manual_seed(1),  
    guidance_scale=3.5,  
).images[0]  

image.save("output.png")