Bug of PixArtAlphaPipeline

[CuddleSabe]

Describe the bug

When I use with torch.autocast("cuda"): images = pipe(...)

the output is noise

Reproduction

import torch from diffusers import PixArtAlphaPipeline

pipe = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-XL-2-1024-MS", torch_dtype=torch.float16) pipe.enable_model_cpu_offload() prompt = "A small cactus with a happy face in the Sahara desert."

with torch.autocast("cuda"): image = pipe(prompt).images[0]

Logs

No response

System Info

master

Who can help?

No response

[tolgacangoz]

It is not recommended to use torch.autocast() when using torch.float16.

[github-actions[bot]]

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the contributing guidelines are likely to be ignored.