How to use Pixart to generate images with different resolutions?

[czg1225]

As shown in the following code, I have set the height and width to 2048 when call the pipeline. However, the output image is still 1024*1024. How can I solve this problem? Thanks.

import torch
from diffusers import PixArtAlphaPipeline

# You can replace the checkpoint id with "PixArt-alpha/PixArt-XL-2-512x512" too.
pipeline = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-XL-2-1024-MS", torch_dtype=torch.float16)
# Enable memory optimizations.
pipeline = pipeline.to("cuda")


torch.manual_seed(42)
torch.cuda.manual_seed_all(42)

prompt = "alien, glowing eyes, crystal body."
image = pipeline(height=2048,width=2048,prompt=prompt,use_resolution_binning=True).images[0]

image.save(f"output.jpg")

[czg1225]

solved