Add Learned PE selection for Auraflow by cloneofsimo · Pull Request #9182 · huggingface/diffusers (original) (raw)

from diffusers import AuraFlowPipeline import torch

pipeline = AuraFlowPipeline.from_pretrained( "/home/ubuntu/auraflow_converter/afv3", torch_dtype=torch.float16, use_safetensors=True, 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]

save

image.save("image.png")