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]
tolgacangoz
commented
2 months ago
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